lucidrains / x-transformersLinks
A concise but complete full-attention transformer with a set of promising experimental features from various papers
☆5,693Updated 3 weeks ago
Alternatives and similar repositories for x-transformers
Users that are interested in x-transformers are comparing it to the libraries listed below
Sorting:
- Vector (and Scalar) Quantization, in Pytorch☆3,732Updated last week
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,174Updated last year
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,290Updated last week
- Structured state space sequence models☆2,783Updated last year
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,329Updated this week
- Pytorch library for fast transformer implementations☆1,753Updated 2 years ago
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,131Updated last week
- Foundation Architecture for (M)LLMs☆3,121Updated last year
- Fast and memory-efficient exact attention☆20,804Updated last week
- TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.☆1,671Updated last week
- maximal update parametrization (µP)☆1,636Updated last year
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆2,896Updated 5 months ago
- Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.☆3,247Updated 6 months ago
- Implementation of Denoising Diffusion Probabilistic Model in Pytorch☆10,228Updated 3 months ago
- Diffusion model papers, survey, and taxonomy☆3,295Updated 2 months ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,181Updated 2 years ago
- PyTorch extensions for high performance and large scale training.☆3,386Updated 7 months ago
- An implementation of Performer, a linear attention-based transformer, in Pytorch☆1,164Updated 3 years ago
- Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.☆1,833Updated 2 weeks ago
- Model interpretability and understanding for PyTorch☆5,475Updated this week
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…☆24,546Updated this week
- An ultimately comprehensive paper list of Vision Transformer/Attention, including papers, codes, and related websites☆4,975Updated last year
- View model summaries in PyTorch!☆2,884Updated last week
- Simple, minimal implementation of the Mamba SSM in one file of PyTorch.☆2,887Updated last year
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆780Updated 4 months ago
- 🚀 Efficient implementations of state-of-the-art linear attention models☆3,937Updated this week
- Reformer, the efficient Transformer, in Pytorch☆2,189Updated 2 years ago
- PyTorch Re-Implementation of "The Sparsely-Gated Mixture-of-Experts Layer" by Noam Shazeer et al. https://arxiv.org/abs/1701.06538☆1,207Updated last year
- Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"☆8,091Updated last year
- A simple and efficient Mamba implementation in pure PyTorch and MLX.☆1,372Updated 11 months ago