A unified, comprehensive and efficient recommendation library
- recommender-systems
- collaborative-filtering
- knowledge-graph
- ctr-prediction
Repository Index
47,133 repositories indexed
97–120 of 192 · Most stars
Scalable and user friendly neural :brain: forecasting algorithms.
- deep-learning
- forecasting
- esrnn
- nbeats
Deepchecks: Tests for Continuous Validation of ML Models & Data. Deepchecks is a holistic open-source solution for all of your AI & ML validation needs, enabling to thoroughly test your data and models from research to production.
- machine-learning
- ml
- model-validation
- data-validation
A highly efficient implementation of Gaussian Processes in PyTorch
- pytorch
- gaussian-processes
- gpu-acceleration
A python library for self-supervised learning and latent world models on images.
- deep-learning
- self-supervised-learning
- machine-learning
- computer-vision
OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.
- reinforcement-learning
- multiagent-reinforcement-learning
- self-play
- imitation-learning
Fast, flexible and easy to use probabilistic modelling in Python.
- python
- machine-learning
- probabilistic-graphical-models
- pytorch
Visual tracking library based on PyTorch.
- computer-vision
- tracking
- machine-learning
- visual-tracking
PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.
Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2
- pytorch
- protein-structure
- alphafold2
Accelerated deep learning R&D
Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks
PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)
- pytorch
- graph-neural-networks
- temporal-networks
- temporal-graphs
Data manipulation and transformation for audio signal processing, powered by PyTorch
- audio
- python
- io
- speech
Multilingual Automatic Speech Recognition with word-level timestamps and confidence
Automatic architecture search and hyperparameter optimization for PyTorch
- automl
- deep-learning
- tabular-data
- pytorch
Reinforcement Learning in PyTorch
An open source library and framework for deep learning on satellite and aerial imagery.
Source code of PyGAD, a Python 3 library for building the genetic algorithm and training machine learning algorithms (Keras & PyTorch).
- python
- genetic-algorithm
- optimization
- numpy
A toolkit for reproducible reinforcement learning research.
- rl-algorithms
- reproducibility
- pytorch
- tensorflow
Transform geospatial relations into graphs for Graph Neural Networks and spatial network analysis
- network-analysis
- spatial-analysis
- geographical-information-system
- graph-neural-networks
Clean PyTorch implementations of imitation and reward learning algorithms
- reward-learning
- inverse-reinforcement-learning
- imitation-learning
- gymnasium
A deep learning toolkit for healthcare AI: EHR (MIMIC, eICU, OMOP), clinical text, signals and imaging. 30+ models, reproducible pipelines.
- healthcare
- data-mining
- deep-learning
- preprocessing
An offline deep reinforcement learning library
- deep-reinforcement-learning
- deep-learning
- pytorch
- offline-rl