Pinned

1. 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 13.2k 2.2k

2. DALLE2-pytorch Public

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

   Python 9.1k 798

3. imagen-pytorch Public

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

   Python 6.4k 590

4. x-transformers Public

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

   Python 2.6k 228

5. musiclm-pytorch Public

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

   Python 1.7k 107

6. alphafold2 Public

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

   Python 1.3k 222