CLAIRE-Labo / flash_attentionLinks
A basic pure pytorch implementation of flash attention
☆16Updated 10 months ago
Alternatives and similar repositories for flash_attention
Users that are interested in flash_attention are comparing it to the libraries listed below
Sorting:
- ☆87Updated last year
- Code for the paper "Function-Space Learning Rates"☆23Updated 2 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆131Updated last year
- 📄Small Batch Size Training for Language Models☆43Updated last week
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- Using FlexAttention to compute attention with different masking patterns☆44Updated 11 months ago
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆56Updated last year
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆56Updated 5 months ago
- ☆85Updated last year
- ☆82Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆100Updated last year
- ☆37Updated last year
- Collection of autoregressive model implementation☆86Updated 4 months ago
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…☆86Updated this week
- supporting pytorch FSDP for optimizers☆84Updated 8 months ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆82Updated 10 months ago
- ☆34Updated 11 months ago
- ☆34Updated last year
- LL3M: Large Language and Multi-Modal Model in Jax☆74Updated last year
- WIP☆94Updated last year
- ☆53Updated last year
- Mixture-of-Transformers: A Sparse and Scalable Architecture for Multi-Modal Foundation Models. TMLR 2025.☆93Updated 3 months ago
- σ-GPT: A New Approach to Autoregressive Models☆67Updated last year
- The evaluation framework for training-free sparse attention in LLMs☆91Updated 2 months ago
- The official github repo for "Diffusion Language Models are Super Data Learners".☆107Updated 3 weeks ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆153Updated 2 months ago
- GoldFinch and other hybrid transformer components☆46Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 2 months ago
- Implementation of Infini-Transformer in Pytorch☆111Updated 7 months ago
- Experiment of using Tangent to autodiff triton☆80Updated last year