Scalable and user friendly neural forecasting algorithms.
Lightning fast C++/CUDA neural network framework
Simulation of spiking neural networks (SNNs) using PyTorch
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
Python-based neural networks API
JAX-based neural network library
Embed images and sentences into fixed-length vectors
TensorFlow-based neural network library
Hummingbird compiles trained ML models into tensor computation
Visualizer for neural network, deep learning, machine learning models
Graph Neural Network Library for PyTorch
Time series forecasting with PyTorch
A cross-platform Python library for differentiable programming
A library for scientific machine learning & physics-informed learning
Training PyTorch models with differential privacy
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Spatiotemporal Signal Processing with Neural Machine Learning Models
Python package built to ease deep learning on graph
Implementation of Imagen, Google's Text-to-Image Neural Network
A python library for easy manipulation and forecasting of time series
JAX-based neural network library
Source code of PyGAD, Python 3 library for building genetic algorithms