Scalable and user friendly neural forecasting algorithms.
AutoML toolkit for automate machine learning lifecycle
Python-based neural networks API
A lightweight audio-to-MIDI converter with pitch bend detection
DeepVariant is an analysis pipeline that uses a deep neural networks
Message Passing Neural Networks for Molecule Property Prediction
Simulation of spiking neural networks (SNNs) using PyTorch
AIMET is a library that provides advanced quantization and compression
Fault-tolerant, highly scalable GPU orchestration
A scikit-learn compatible neural network library that wraps PyTorch
A cross-platform Python library for differentiable programming
TensorFlow-based neural network library
Spatiotemporal Signal Processing with Neural Machine Learning Models
Python package built to ease deep learning on graph
JAX-based neural network library
Hummingbird compiles trained ML models into tensor computation
Embed images and sentences into fixed-length vectors
Implementation of DeepLabCut
Fast image augmentation library and an easy-to-use wrapper
A library for scientific machine learning & physics-informed learning
Implementation of Imagen, Google's Text-to-Image Neural Network
Graph Neural Network Library for PyTorch
Python library for Representation Learning on Knowledge Graphs
Source code of PyGAD, Python 3 library for building genetic algorithms
OpenMMLab Model Deployment Framework