Scalable and user friendly neural forecasting algorithms.
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
Message Passing Neural Networks for Molecule Property Prediction
Open source machine learning framework
JAX-based neural network library
Python-based neural networks API
TensorFlow-based neural network library
Embed images and sentences into fixed-length vectors
Visualizer for neural network, deep learning, machine learning models
Hummingbird compiles trained ML models into tensor computation
OpenMMLab Model Deployment Framework
Graph Neural Network Library for PyTorch
AIMET is a library that provides advanced quantization and compression
A cross-platform Python library for differentiable programming
A library for scientific machine learning & physics-informed learning
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Time series forecasting with PyTorch
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
Core ML tools contain supporting tools for Core ML model conversion
JAX-based neural network library
Training PyTorch models with differential privacy