Geometric deep learning extension library for PyTorch
PyTorch extensions for fast R&D prototyping and Kaggle farming
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
A Python package for extending the official PyTorch
Simplest working implementation of Stylegan2
High quality, fast, modular reference implementation of SSD in PyTorch
Automatic architecture search and hyperparameter optimization
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
Spatiotemporal Signal Processing with Neural Machine Learning Models
High-level training, data augmentation, and utilities for Pytorch
Transfer Learning Library for Domain Adaptation, Task Adaptation, etc.
Open-source pre-training implementation of Google's LaMDA in PyTorch
Integrating the Best of TF into PyTorch, for Machine Learning