Scalable and user friendly neural forecasting algorithms.
Simulation of spiking neural networks (SNNs) using PyTorch
Lightning fast C++/CUDA neural network framework
Message Passing Neural Networks for Molecule Property Prediction
A scikit-learn compatible neural network library that wraps PyTorch
Open source machine learning framework
OpenMMLab Model Deployment Framework
JAX-based neural network library
Embed images and sentences into fixed-length vectors
Python-based neural networks API
Hummingbird compiles trained ML models into tensor computation
TensorFlow-based neural network library
Visualizer for neural network, deep learning, machine learning models
DeepVariant is an analysis pipeline that uses a deep neural networks
Graph Neural Network Library for PyTorch
Time series forecasting with PyTorch
A cross-platform Python library for differentiable programming
Spatiotemporal Signal Processing with Neural Machine Learning Models
A library for scientific machine learning & physics-informed learning
Training PyTorch models with differential privacy
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Python package built to ease deep learning on graph
Implementation of Imagen, Google's Text-to-Image Neural Network
JAX-based neural network library
AIMET is a library that provides advanced quantization and compression