lucidrains / x-transformersView external linksLinks
A concise but complete full-attention transformer with a set of promising experimental features from various papers
☆5,800Feb 7, 2026Updated last week
Alternatives and similar repositories for x-transformers
Users that are interested in x-transformers are comparing it to the libraries listed below
Sorting:
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,395Jan 26, 2026Updated 2 weeks ago
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…☆24,993Updated this week
- Vector (and Scalar) Quantization, in Pytorch☆3,870Updated this week
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,336Feb 5, 2026Updated last week
- Fast and memory-efficient exact attention☆22,231Updated this week
- Implementation of Denoising Diffusion Probabilistic Model in Pytorch☆10,455Aug 4, 2025Updated 6 months ago
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,491Feb 6, 2026Updated last week
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆22,021Jan 23, 2026Updated 3 weeks ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,351Updated this week
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,184Nov 27, 2024Updated last year
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,823Feb 4, 2026Updated last week
- An open source implementation of CLIP.☆13,383Updated this week
- Pytorch library for fast transformer implementations☆1,761Mar 23, 2023Updated 2 years ago
- 🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.☆32,768Updated this week
- Reformer, the efficient Transformer, in Pytorch☆2,193Jun 21, 2023Updated 2 years ago
- ☆7,291Jul 2, 2024Updated last year
- Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch☆8,410Oct 7, 2024Updated last year
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)…☆14,351Updated this week
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆5,629Feb 17, 2024Updated last year
- Facebook AI Research Sequence-to-Sequence Toolkit written in Python.☆32,143Sep 30, 2025Updated 4 months ago
- Train transformer language models with reinforcement learning.☆17,360Updated this week
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,194Aug 22, 2023Updated 2 years ago
- Accessible large language models via k-bit quantization for PyTorch.☆7,952Updated this week
- Taming Transformers for High-Resolution Image Synthesis☆6,429Jul 30, 2024Updated last year
- PyTorch extensions for high performance and large scale training.☆3,397Apr 26, 2025Updated 9 months ago
- A concise but complete implementation of CLIP with various experimental improvements from recent papers☆722Oct 16, 2023Updated 2 years ago
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆41,578Feb 7, 2026Updated last week
- Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"☆8,352May 31, 2024Updated last year
- Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch☆11,332May 11, 2024Updated last year
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆23,494Aug 15, 2024Updated last year
- Foundation Architecture for (M)LLMs☆3,130Apr 11, 2024Updated last year
- Implementation of RETRO, Deepmind's Retrieval based Attention net, in Pytorch☆879Oct 30, 2023Updated 2 years ago
- Ongoing research training transformer models at scale☆15,162Updated this week
- Transformer related optimization, including BERT, GPT☆6,392Mar 27, 2024Updated last year
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆804Jan 30, 2026Updated 2 weeks ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆827May 5, 2024Updated last year
- 🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.☆20,619Updated this week
- CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image☆32,562Jul 23, 2024Updated last year
- Development repository for the Triton language and compiler☆18,387Updated this week