lucidrains / ring-attention-pytorch
Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch
☆506Updated 4 months ago
Alternatives and similar repositories for ring-attention-pytorch:
Users that are interested in ring-attention-pytorch are comparing it to the libraries listed below
- Large Context Attention☆690Updated last month
- Helpful tools and examples for working with flex-attention☆695Updated this week
- Ring attention implementation with flash attention☆711Updated 3 weeks ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆524Updated last month
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆232Updated 2 weeks ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆447Updated last month
- LLM KV cache compression made easy☆440Updated this week
- Memory optimization and training recipes to extrapolate language models' context length to 1 million tokens, with minimal hardware.☆705Updated 5 months ago
- ☆393Updated 2 weeks ago
- This repository contains the experimental PyTorch native float8 training UX☆222Updated 7 months ago
- Explorations into some recent techniques surrounding speculative decoding☆248Updated 3 months ago
- Muon optimizer: +>30% sample efficiency with <3% wallclock overhead☆505Updated 2 weeks ago
- Implementation of ST-Moe, the latest incarnation of MoE after years of research at Brain, in Pytorch☆318Updated 9 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆223Updated last month
- [ICML 2024] CLLMs: Consistency Large Language Models☆386Updated 4 months ago
- Microsoft Automatic Mixed Precision Library☆581Updated 5 months ago
- Triton-based implementation of Sparse Mixture of Experts.☆207Updated 3 months ago
- Pipeline Parallelism for PyTorch☆757Updated 7 months ago
- ☆218Updated 9 months ago
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆223Updated last month
- Efficient LLM Inference over Long Sequences☆365Updated last month
- [NeurIPS 2024] KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantization☆336Updated 7 months ago
- [ICML 2024] Break the Sequential Dependency of LLM Inference Using Lookahead Decoding☆1,216Updated 2 weeks ago
- Annotated version of the Mamba paper☆475Updated last year
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆207Updated 7 months ago
- [ICLR 2025] DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Heads☆435Updated last month
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆276Updated this week
- ☆140Updated last year
- Megatron's multi-modal data loader☆181Updated this week
- ring-attention experiments☆127Updated 5 months ago