The official implementation for [NeurIPS2025 Oral] Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Free
β975Dec 20, 2025Updated 7 months ago
Alternatives and similar repositories for gated_attention
Users that are interested in gated_attention are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- PyTorch implementation of JiT https://arxiv.org/abs/2511.13720β2,490Dec 8, 2025Updated 8 months ago
- π Efficient implementations for emerging model architecturesβ5,551Updated this week
- Official PyTorch Implementation of "Diffusion Transformers with Representation Autoencoders"β1,988Feb 25, 2026Updated 5 months ago
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Ruleβ646Mar 13, 2026Updated 4 months ago
- Muon is an optimizer for hidden layers in neural networksβ2,779May 24, 2026Updated 2 months ago
- GPU virtual machines on DigitalOcean Gradient AI β’ AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- β253Nov 19, 2025Updated 8 months ago
- β1,568Nov 17, 2025Updated 8 months ago
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruningβ152Feb 25, 2026Updated 5 months ago
- β3,466Mar 17, 2026Updated 4 months ago
- Code release for paper "Test-Time Training Done Right"β501Jan 5, 2026Updated 7 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkβ1,696Mar 16, 2025Updated last year
- [ICLR 2025] When Attention Sink Emerges in Language Models: An Empirical View (Spotlight)β165Jul 8, 2025Updated last year
- π₯ A minimal training framework for scaling FLA modelsβ410Apr 22, 2026Updated 3 months ago
- π³ Efficient Triton implementations for "Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attention"β1,018Feb 5, 2026Updated 6 months ago
- Deploy on Railway without the complexity - Free Credits Offer β’ AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- [NeurIPS 2025] An official implementation of Flow-GRPO: Training Flow Matching Models via Online RLβ2,466May 7, 2026Updated 3 months ago
- Official PyTorch implementation for "Large Language Diffusion Models"β3,933Jul 15, 2026Updated 3 weeks ago
- [ICCV 2025] Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformersβ513Dec 6, 2025Updated 8 months ago
- Official Implemenation for RAEv2: Improved Baselines with Representation Autoencodersβ317May 21, 2026Updated 2 months ago
- PyTorch implementation of NEPAβ340Feb 9, 2026Updated 6 months ago
- [CVPR 2026] SpatialVID: A Large-Scale Video Dataset with Spatial Annotationsβ595Apr 22, 2026Updated 3 months ago
- A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attentionβ307Dec 1, 2025Updated 8 months ago
- [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Modelsβ1,521Dec 16, 2025Updated 7 months ago
- [ECCV 2026] Towards Scalable Pre-training of Visual Tokenizers for Generationβ500Apr 15, 2026Updated 3 months ago
- End-to-end encrypted email - Proton Mail β’ AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- implementations and experimentation on mHC by deepseek - https://arxiv.org/abs/2512.24880β372Feb 17, 2026Updated 5 months ago
- Cameras as Relative Positional Encodingβ746Dec 18, 2025Updated 7 months ago
- β285Jun 6, 2025Updated last year
- [ICML 2026] Official codebase for "Causal Forcing: Autoregressive Diffusion Distillation Done Right for High-Quality Real-Time Interactivβ¦β920Jul 23, 2026Updated 3 weeks ago
- Official implementation of "Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decoding"β1,072May 30, 2026Updated 2 months ago
- rCM & Causal-rCM: Leading and Unified Algorithms/Infrastructures for Bidirectional/Autoregressive Video Diffusion Distillation at Scaleβ780Jun 25, 2026Updated last month
- [ICLR 2026] Streaming 4D Visual Geometry Transformerβ953Oct 27, 2025Updated 9 months ago
- A Distributed Attention Towards Linear Scalability for Ultra-Long Context, Heterogeneous Data Trainingβ907Updated this week
- Physics of Language Models: Part 4.2, Canon Layers at Scale where Synthetic Pretraining Resonates in Realityβ357May 20, 2026Updated 2 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer β’ AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Reference PyTorch implementation and models for DINOv3β11,161Jul 15, 2026Updated 3 weeks ago
- Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)β726Sep 24, 2025Updated 10 months ago
- [ICML'26] Code and website for Self-Flow: Self-Supervised Flow Matching for Scalable Multi-Modal Synthesisβ730May 23, 2026Updated 2 months ago
- MMaDA - Open-Sourced Multimodal Large Diffusion Language Models (dLLMs with block diffusion, mixed-CoT, unified RL)β1,663Feb 14, 2026Updated 5 months ago
- [ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.β1,027Feb 25, 2026Updated 5 months ago
- Scaled diffusion transformer for text-to-speech synthesis (DiT + T5Gemma2 conditioning, TorchTitan & Megatron backends, tested up to 1024β¦β24Mar 29, 2026Updated 4 months ago
- UniGeo: Taming Video Diffusion for Unified Consistent Geometry Estimationβ136Jun 10, 2025Updated last year