Pinned

1. vit-pytorch vit-pytorch Public

   Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

   Python 15.6k 2.5k

2. DALLE2-pytorch DALLE2-pytorch Public

   Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

   Python 10.2k 982

3. imagen-pytorch imagen-pytorch Public

   Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch

   Python 7.4k 689

4. x-transformers x-transformers Public

   A simple but complete full-attention transformer with a set of promising experimental features from various papers

   Python 3.5k 302

5. musiclm-pytorch musiclm-pytorch Public

   Implementation of MusicLM, Google's new SOTA model for music generation using attention networks, in Pytorch

   Python 2.7k 222

6. alphafold2 alphafold2 Public

   To eventually become an unofficial Pytorch implementation / replication of Alphafold2, as details of the architecture get released

   Python 1.4k 243