Open source machine learning framework
Geometric deep learning extension library for PyTorch
C++ Implementation of PyTorch Tutorials for Everyone
PyTorch extensions for fast R&D prototyping and Kaggle farming
Enabling PyTorch on Google TPU
Time series forecasting with PyTorch
Library to help with training and evaluating neural networks
The lightweight PyTorch wrapper for high-performance AI research
Implementation of Imagen, Google's Text-to-Image Neural Network
Implementation of MusicLM music generation model in Pytorch
Implementation of Phenaki Video, which uses Mask GIT
Implementation of AudioLM audio generation model in Pytorch
Implementation of NÜWA, attention network for text to video synthesis
Differentiable SDE solvers with GPU support and efficient sensitivity
Audio generation using diffusion models, in PyTorch
Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis
High quality, fast, modular reference implementation of SSD in PyTorch
A Python package for extending the official PyTorch
Simplest working implementation of Stylegan2
Implementation of Video Diffusion Models
Implementation of Make-A-Video, new SOTA text to video generator
Implementation / replication of DALL-E, OpenAI's Text to Image
Automatic architecture search and hyperparameter optimization
Spatiotemporal Signal Processing with Neural Machine Learning Models
High-level training, data augmentation, and utilities for Pytorch