JAX-based neural network library
A cross-platform Python library for differentiable programming
Hummingbird compiles trained ML models into tensor computation
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Implementation of Imagen, Google's Text-to-Image Neural Network
A library for scientific machine learning & physics-informed learning
A lightweight audio-to-MIDI converter with pitch bend detection
A scikit-learn compatible neural network library that wraps PyTorch
Sandbox for training deep learning networks
BitNet: Scaling 1-bit Transformers for Large Language Models
The Python Code Tutorials
Training PyTorch models with differential privacy
Easy-to-use,Modular and Extendible package of deep-learning models
Graph Neural Network Library for PyTorch
Package of deep-learning based CTR models
Machine learning image inpainting task that removes watermarks
High-Performance Face Recognition Library on PaddlePaddle & PyTorch
Simulation of spiking neural networks (SNNs) using PyTorch
AIMET is a library that provides advanced quantization and compression
Implementation of 'lightweight' GAN, proposed in ICLR 2021
JAX-based neural network library
AI discovers 520000 stable inorganic crystal structures for research
A toolkit to optimize ML models for deployment for Keras & TensorFlow
Core ML tools contain supporting tools for Core ML model conversion
Time series forecasting with PyTorch