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