The official implementation for [NeurIPS2025 Oral] Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Free
β986Dec 20, 2025Updated 9 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,550Dec 8, 2025Updated 10 months ago
- π Efficient implementations for emerging model architecturesβ5,850Updated this week
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Ruleβ680Mar 13, 2026Updated 6 months ago
- Official PyTorch Implementation of "Diffusion Transformers with Representation Autoencoders"β2,038Feb 25, 2026Updated 7 months ago
- Muon is an optimizer for hidden layers in neural networksβ2,872May 24, 2026Updated 4 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.
- β255Nov 19, 2025Updated 10 months ago
- β1,615Nov 17, 2025Updated 10 months ago
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruningβ154Feb 25, 2026Updated 7 months ago
- β3,519Mar 17, 2026Updated 6 months ago
- Code release for paper "Test-Time Training Done Right"β529Jan 5, 2026Updated 9 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkβ1,722Mar 16, 2025Updated last year
- [ICLR 2025] When Attention Sink Emerges in Language Models: An Empirical View (Spotlight)β164Jul 8, 2025Updated last year
- π₯ A minimal training framework for scaling FLA modelsβ421Apr 22, 2026Updated 5 months ago
- π³ Efficient Triton implementations for "Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attention"β1,023Feb 5, 2026Updated 8 months ago
- Simple, predictable pricing with DigitalOcean hosting β’ AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- [NeurIPS 2025] An official implementation of Flow-GRPO: Training Flow Matching Models via Online RLβ2,548May 7, 2026Updated 5 months ago
- Official PyTorch implementation for "Large Language Diffusion Models"β3,995Jul 15, 2026Updated 2 months ago
- [ICCV 2025] Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformersβ524Dec 6, 2025Updated 10 months ago
- Official Implemenation for RAEv2: Improved Baselines with Representation Autoencodersβ330May 21, 2026Updated 4 months ago
- PyTorch implementation of NEPAβ343Feb 9, 2026Updated 8 months ago
- [CVPR 2026] SpatialVID: A Large-Scale Video Dataset with Spatial Annotationsβ604Apr 22, 2026Updated 5 months ago
- A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attentionβ312Dec 1, 2025Updated 10 months ago
- [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Modelsβ1,545Dec 16, 2025Updated 9 months ago
- [ECCV 2026] Towards Scalable Pre-training of Visual Tokenizers for Generationβ509Apr 15, 2026Updated 5 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.
- implementations and experimentation on mHC by deepseek - https://arxiv.org/abs/2512.24880β378Feb 17, 2026Updated 7 months ago
- Cameras as Relative Positional Encodingβ761Dec 18, 2025Updated 9 months ago
- β288Jun 6, 2025Updated last year
- Official implementation of "Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decoding"β1,095May 30, 2026Updated 4 months ago
- [ICML 2026] Official codebase for "Causal Forcing: Autoregressive Diffusion Distillation Done Right for High-Quality Real-Time Interactivβ¦β1,002Oct 3, 2026Updated last week
- rCM & Causal-rCM: Leading and Unified Algorithms/Infrastructures for Bidirectional/Autoregressive Video Diffusion Distillation at Scaleβ819Jun 25, 2026Updated 3 months ago
- [ICLR 2026] Streaming 4D Visual Geometry Transformerβ982Oct 27, 2025Updated 11 months ago
- A Distributed Attention Towards Linear Scalability for Ultra-Long Context, Heterogeneous Data Trainingβ951Updated this week
- Physics of Language Models: Part 4.2, Canon Layers at Scale where Synthetic Pretraining Resonates in Realityβ362Aug 21, 2026Updated last month
- 1-Click AI Models by DigitalOcean Gradient β’ AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Reference PyTorch implementation and models for DINOv3β11,527Jul 15, 2026Updated 2 months ago
- Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)β729Sep 24, 2025Updated last year
- MMaDA - Open-Sourced Multimodal Large Diffusion Language Models (dLLMs with block diffusion, mixed-CoT, unified RL)β1,674Feb 14, 2026Updated 7 months ago
- [ICML'26] Code and website for Self-Flow: Self-Supervised Flow Matching for Scalable Multi-Modal Synthesisβ776May 23, 2026Updated 4 months ago
- Scaled diffusion transformer for text-to-speech synthesis (DiT + T5Gemma2 conditioning, TorchTitan & Megatron backends, tested up to 1024β¦β25Mar 29, 2026Updated 6 months ago
- UniGeo: Taming Video Diffusion for Unified Consistent Geometry Estimationβ136Jun 10, 2025Updated last year
- [ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.β1,260Feb 25, 2026Updated 7 months ago