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