Repository Index

An autoML framework & toolkit for machine learning on graphs.

• graph-neural-networks
• automl
• pytorch
• pytorch-geometric

A lightweight header-only library for using Keras (TensorFlow) models in C++.

• tensorflow
• deep-learning
• keras
• cpp

OpenAI Gym environments for an open-source quadruped robot (SpotMicro)

• openai-gym
• openai-gym-environments
• robotics
• tensorflow

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments

• reinforcement-learning
• python
• deep-learning
• gym

fastFM: A Library for Factorization Machines

• machine-learning
• recommender-system
• factorization-machines
• matrix-factorization

mlr3: Machine Learning in R - next generation

• machine-learning
• data-science
• classification
• regression

Curated List: Practical Natural Language Processing done in Ruby

• machine-learning
• natural-language-processing
• ruby
• nlp

A curated list of gradient boosting research papers with implementations.

• gradient-boosting
• gradient-boosting-classifier
• gradient-boosting-machine
• gradient-boosted-trees

This Word Does Not Exist

• machine-learning
• gpt-2
• transformers
• natural-language-processing

[CONTRIBUTORS WELCOME] Generalized Additive Models in Python

• machine-learning
• data-science
• scientific-computing
• python

Python package for graph statistics

• graph
• data-science
• networks
• python

allRank is a framework for training learning-to-rank neural models based on PyTorch.

• learning-to-rank
• ndcg
• ranking
• information-retrieval

Examples and experiments around ML for upcoming Coding Train videos

Pretrained model hub for Keras 3.

• deep-learning
• keras
• machine-learning
• nlp

Molecular Sets (MOSES): A Benchmarking Platform for Molecular Generation Models

• machine-learning
• generative-models
• drug-discovery
• benchmark

Sparkling Water provides H2O functionality inside Spark cluster

• h2o
• spark
• machine-learning
• integration

R interface for Apache Spark

• r
• rstats
• apache-spark
• machine-learning

A scikit-learn based module for multi-label et. al. classification

• classification
• scikit
• scikit-learn
• multi-label

Combines the ease of use of scikit-learn with the power of Theano/Lasagne

• scikit-learn
• lasagne
• deep-learning
• machine-learning

🔧 A curated list of awesome dataset tools

• datasets
• awsome
• awsome-list
• machine-learning

Tiny YOLO for iOS implemented using CoreML but also using the new MPS graph API.

• core-ml
• mps
• metal
• machine-learning

A uniform interface to run deep learning models from multiple frameworks

• tensorflow
• pytorch
• keras
• deep-learning

Rumale is a machine learning library in Ruby

• ruby
• machine-learning
• data-science
• data-analysis

onnx-go gives the ability to import a pre-trained neural network within Go without being linked to a framework or library.

• go
• onnx
• neural-network
• open-source