Stable ABI FlashMLA Fork Released for PyTorch

ezyang · x · 2026-08-18

A fork of DeepSeek's FlashMLA introduces LibTorch and CPython ABI stability, allowing a single wheel build to run across Python 3.10+ and PyTorch 2.10+. The project supports direct installation from GitHub and requires CUDA 13.0+, targeting improved compatibility for efficient multi-head latent attention kernels.

Original post →

More from Research

Research channel →