Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Python 25k 3.5k
Implementation of Alphafold 3 from Google Deepmind in Pytorch
Python 1.6k 215
Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
Python 8.4k 797
A concise but complete full-attention transformer with a set of promising experimental features from various papers
Python 5.8k 504
Vector (and Scalar) Quantization, in Pytorch
Python 3.9k 318
Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI
Python 1.3k 71
