Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Python 25.3k 3.5k
Implementation of Alphafold 3 from Google Deepmind in Pytorch
Python 1.7k 227
Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
Python 8.4k 800
A concise but complete full-attention transformer with a set of promising experimental features from various papers
Python 5.9k 513
Vector (and Scalar) Quantization, in Pytorch
Python 4k 330
Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI
Python 1.4k 73
