Implementation of RETRO, Deepmind's Retrieval based Attention net, in Pytorch
APACHE-2.0 License
DALL·E Mini - Generate images from a text prompt
Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architectu...
Implementation of NÜWA, state of the art attention network for text to video synthesis, in Pytorch
A concise but complete implementation of CLIP with various experimental improvements from recent ...
Implementation of SoundStorm, Efficient Parallel Audio Generation from Google Deepmind, in Pytorch
Implementation of Phenaki Video, which uses Mask GIT to produce text guided videos of up to 2 min...
Implementation of Muse: Text-to-Image Generation via Masked Generative Transformers, in Pytorch
Implementation of Memorizing Transformers (ICLR 2022), attention net augmented with indexing and ...
Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and...
Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-E...
An implementation of Performer, a linear attention-based transformer, in Pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with ...
Implementation of AudioLM, a SOTA Language Modeling Approach to Audio Generation out of Google Re...
Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch
A simple but complete full-attention transformer with a set of promising experimental features fr...