fal-ai-community / NativeSparseAttention
research impl of Native Sparse Attention (2502.11089)
☆54Updated last month
Alternatives and similar repositories for NativeSparseAttention:
Users that are interested in NativeSparseAttention are comparing it to the libraries listed below
- supporting pytorch FSDP for optimizers☆80Updated 3 months ago
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆44Updated 2 weeks ago
- Focused on fast experimentation and simplicity☆70Updated 3 months ago
- ☆76Updated 8 months ago
- ☆91Updated 2 months ago
- ☆33Updated 6 months ago
- https://x.com/BlinkDL_AI/status/1884768989743882276☆27Updated last month
- ☆59Updated 4 months ago
- ☆21Updated 4 months ago
- GoldFinch and other hybrid transformer components☆45Updated 8 months ago
- ☆47Updated this week
- ☆24Updated 3 months ago
- Collection of autoregressive model implementation☆83Updated last month
- ☆19Updated this week
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers☆87Updated 8 months ago
- Explorations into the recently proposed Taylor Series Linear Attention☆95Updated 7 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆91Updated 3 weeks ago
- ☆27Updated last year
- Implementation of GateLoop Transformer in Pytorch and Jax☆87Updated 9 months ago
- Experiment of using Tangent to autodiff triton☆78Updated last year
- Combining SOAP and MUON☆13Updated last month
- ☆53Updated last year
- WIP☆93Updated 7 months ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆121Updated 7 months ago
- ☆95Updated 10 months ago
- Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.☆47Updated last week
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated last week
- Triton Implementation of HyperAttention Algorithm☆47Updated last year
- DPO, but faster 🚀☆40Updated 3 months ago
- DeMo: Decoupled Momentum Optimization☆185Updated 3 months ago