lucidrains / autoregressive-linear-attention-cudaView external linksLinks
CUDA implementation of autoregressive linear attention, with all the latest research findings
☆46May 23, 2023Updated 2 years ago
Alternatives and similar repositories for autoregressive-linear-attention-cuda
Users that are interested in autoregressive-linear-attention-cuda are comparing it to the libraries listed below
Sorting:
- Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk☆47Jul 16, 2023Updated 2 years ago
- Toy genetic algorithm in Pytorch☆55Apr 29, 2025Updated 9 months ago
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆123Oct 17, 2024Updated last year
- Yet another random morning idea to be quickly tried and architecture shared if it works; to allow the transformer to pause for any amount…☆53Oct 22, 2023Updated 2 years ago
- Explorations into the recently proposed Taylor Series Linear Attention☆100Aug 18, 2024Updated last year
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- Implementation of 🌻 Mirasol, SOTA Multimodal Autoregressive model out of Google Deepmind, in Pytorch☆91Dec 22, 2023Updated 2 years ago
- Implementation of the transformer proposed in "Building Blocks for a Complex-Valued Transformer Architecture"☆88Oct 13, 2023Updated 2 years ago
- Implementation of a holodeck, written in Pytorch☆18Nov 1, 2023Updated 2 years ago
- A library for squeakily cleaning and filtering language datasets.☆50Jul 10, 2023Updated 2 years ago
- Utilities for PyTorch distributed☆25Feb 27, 2025Updated 11 months ago
- ☆23Jun 18, 2024Updated last year
- ☆21Mar 3, 2025Updated 11 months ago
- Implementation of the Llama architecture with RLHF + Q-learning☆170Feb 1, 2025Updated last year
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆220Feb 13, 2023Updated 3 years ago
- Code repository for the public reproduction of the language modelling experiments on "MatFormer: Nested Transformer for Elastic Inference…☆31Nov 14, 2023Updated 2 years ago
- ☆12Jan 4, 2024Updated 2 years ago
- Build modern UIs in Jupyter with Python☆12Dec 28, 2022Updated 3 years ago
- RWKV-7 mini☆12Mar 29, 2025Updated 10 months ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆135Oct 15, 2025Updated 4 months ago
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆207Feb 14, 2024Updated 2 years ago
- [EMNLP 2023] Official implementation of the algorithm ETSC: Exact Toeplitz-to-SSM Conversion our EMNLP 2023 paper - Accelerating Toeplitz…☆14Oct 17, 2023Updated 2 years ago
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆19Oct 9, 2022Updated 3 years ago
- ANE accelerated embedding models!☆20Dec 11, 2024Updated last year
- ☆34Sep 10, 2024Updated last year
- Implementation of 2-simplicial attention proposed by Clift et al. (2019) and the recent attempt to make practical in Fast and Simplex, Ro…☆46Sep 2, 2025Updated 5 months ago
- Implementation of "PaLM2-VAdapter:" from the multi-modal model paper: "PaLM2-VAdapter: Progressively Aligned Language Model Makes a Stron…☆17Nov 11, 2024Updated last year
- https://hf.co/hexgrad/Kokoro-82M☆14Jan 14, 2026Updated last month
- Implementation of Agent Attention in Pytorch☆93Jul 10, 2024Updated last year
- The 2D discrete wavelet transform for JAX☆45Feb 28, 2023Updated 2 years ago
- ☆19Dec 1, 2021Updated 4 years ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆103Dec 22, 2024Updated last year
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆207Aug 26, 2023Updated 2 years ago
- Simple python library for generating your own perfetto traces for your application. Can be used for both app instrumentation and custom …☆24Jun 22, 2025Updated 7 months ago
- Hacks for PyTorch☆19Apr 18, 2023Updated 2 years ago
- Learning disentangled representations of single-cell data for high-quality generation☆16Mar 16, 2024Updated last year
- ☆24Dec 11, 2024Updated last year
- A Framework aims to wisely initialize unseen subword embeddings in PLMs for efficient large-scale continued pretraining☆18Nov 26, 2023Updated 2 years ago
- Simple Autogpt with tree of thoughts☆14May 25, 2023Updated 2 years ago