Back to Gap Map Infrastructure / Compilers & Model Optimization

mLSTM Kernels

NXAI
source available / Overall score: 2.0

mLSTM Kernels (Tiled Flash Linear Attention) provides fast, chunkwise-parallel Triton kernels for the mLSTM used in xLSTM, with JAX and PyTorch native implementations and Triton integrations for both. NXAI, the xLSTM team's company, distributes it under its own restrictive community license rather than an OSI-approved one.

Openness

2 high confidence
2.0
license
NXAI-Community-License(non-OSI, Llama-3-style base with a revenue-bounded commercial term - a separate license is required above EUR 100,000,000 in consolidated annual revenue)
source
public
core features withheld
no

NX-AI publishes mlstm_kernels under its own NXAI Community License rather than an OSI-approved one, requiring a separate, NX-AI-granted license once a licensee's annual revenue exceeds roughly EUR 100 million. The repository itself is public and unarchived with no separate withheld component, so it is the license term itself, not gated code, that keeps this below full open source.

  • https://api.github.com/repos/NX-AI/mlstm_kernels recorded 2026-09-02

    Repo metadata for mlstm-kernels's canonical repository - private false, archived false, fork false.

  • https://github.com/NX-AI/mlstm_kernels/blob/main/LICENSE recorded 2026-10-07

    LICENSE body is the NXAI Community License Agreement. Section 2, 'Additional Commercial Terms', requires a separate commercial license, granted at NXAI's sole discretion, once a licensee's consolidated annual revenue exceeds EUR 100,000,000 and it incorporates the material into a commercial product or service - a revenue-bounded commercial restriction, not an unconditional grant.

  • https://github.com/NX-AI/mlstm_kernels/blob/main/README.md recorded 2026-10-07

    README describes the library as providing fast, efficient mLSTM training and inference Triton kernels with JAX, PyTorch and Triton implementations, and cites the NXAI Community License Agreement governing the repository.

Adoption

2 high confidence
2.0

PyPI downloads of `mlstm-kernels` are the adoption signal here; its README documents `pip install mlstm_kernels` as the install path, and the package is published by the NX-AI team that maintains the project.

Capability

2 medium confidence
2.0

mLSTM Kernels provides chunkwise-parallel Triton kernels for one recurrent architecture family, mLSTM/xLSTM, with JAX and PyTorch integrations. That is a narrow, single-architecture kernel set, well short of the broad, production-grade engine TensorRT represents in this category.

  • https://github.com/NX-AI/mlstm_kernels/blob/main/README.md recorded 2026-10-07

    README describes the library as providing fast, efficient mLSTM training and inference Triton kernels with JAX, PyTorch and Triton implementations, and cites the NXAI Community License Agreement governing the repository.

Verified 2026-09-24