AutoML toolkit for automate machine learning lifecycle
Lightning fast C++/CUDA neural network framework
Simulation of spiking neural networks (SNNs) using PyTorch
Scalable and user friendly neural forecasting algorithms.
A scikit-learn compatible neural network library that wraps PyTorch
JAX-based neural network library
TensorFlow-based neural network library
Visualizer for neural network, deep learning, machine learning models
Hummingbird compiles trained ML models into tensor computation
Implementation of Imagen, Google's Text-to-Image Neural Network
A cross-platform Python library for differentiable programming
A library for scientific machine learning & physics-informed learning
OpenMMLab Model Deployment Framework
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Message Passing Neural Networks for Molecule Property Prediction
AIMET is a library that provides advanced quantization and compression
Time series forecasting with PyTorch
DeepVariant is an analysis pipeline that uses a deep neural networks
Graph Neural Network Library for PyTorch
oneAPI Deep Neural Network Library (oneDNN)
Training PyTorch models with differential privacy
Core ML tools contain supporting tools for Core ML model conversion
JAX-based neural network library
Open source machine learning framework
Decentralized deep learning in PyTorch. Built to train models