AIMET is a library that provides advanced quantization and compression
Implementation of Imagen, Google's Text-to-Image Neural Network
BitNet: Scaling 1-bit Transformers for Large Language Models
A lightweight audio-to-MIDI converter with pitch bend detection
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
JAX-based neural network library
A library for scientific machine learning & physics-informed learning
A scikit-learn compatible neural network library that wraps PyTorch
An AI for Music Generation
Hummingbird compiles trained ML models into tensor computation
Graph Neural Network Library for PyTorch
Package of deep-learning based CTR models
Sandbox for training deep learning networks
The Python Code Tutorials
Training PyTorch models with differential privacy
Easy-to-use,Modular and Extendible package of deep-learning models
Time series forecasting with PyTorch
A cross-platform Python library for differentiable programming
Decentralized deep learning in PyTorch. Built to train models
DeepVariant is an analysis pipeline that uses a deep neural networks
Simulation of spiking neural networks (SNNs) using PyTorch
A refreshing functional take on deep learning
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Machine learning image inpainting task that removes watermarks
JAX-based neural network library