A developer has open-sourced a Decay-Gated O(N) Causal Linear Attention architecture featuring fused Triton/CUDA state accumulation kernels designed to bypass quadratic MHA bottlenecks.
- Context Scaling: 128K context sequence executed in 108.78ms with <185MB VRAM on NVIDIA T4.
- Kernel Acceleration: Fused Triton kernel delivers a 3.60x speedup over standard PyTorch implementation.
- Precision Guard: Uses FP32 Master Accumulator + RMS Normalization to prevent FP16/BF16 gradient overflow during long-context training.
- Retrieval Accuracy: Passed 16K Needle-In-A-Haystack (NIAH) benchmark with 1.89x SNR.
The author is seeking feedback on integrating the model into Hugging Face Transformers or conducting benchmark comparisons with existing linear models.