Skip to content
@NX-AI

NXAI

Pinned Loading

  1. xlstm xlstm Public

    Official repository of the xLSTM.

    Python 2k 156

  2. xlstm-jax xlstm-jax Public

    Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.co/NX-AI/xLSTM-7b.

    Python 104 4

  3. mlstm_kernels mlstm_kernels Public

    Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.

    Jupyter Notebook 70 6

  4. tirex tirex Public

    TiRex: Zero-Shot Forecasting Across Long and Short Horizons with Enhanced In-Context Learning

    Python 179 27

Repositories

Showing 10 of 10 repositories
  • xlstm_scaling_laws Public

    Code and data to explore neural scaling laws of xLSTM and Transformer models.

    NX-AI/xlstm_scaling_laws’s past year of commit activity
    Jupyter Notebook 4 0 0 0 Updated Oct 3, 2025
  • tirex Public

    TiRex: Zero-Shot Forecasting Across Long and Short Horizons with Enhanced In-Context Learning

    NX-AI/tirex’s past year of commit activity
    Python 179 27 3 0 Updated Oct 2, 2025
  • transformers Public Forked from huggingface/transformers

    🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

    NX-AI/transformers’s past year of commit activity
    Python 1 Apache-2.0 31,116 0 0 Updated Sep 22, 2025
  • vllm Public Forked from vllm-project/vllm

    A high-throughput and memory-efficient inference and serving engine for LLMs

    NX-AI/vllm’s past year of commit activity
    Python 0 Apache-2.0 10,623 0 0 Updated Sep 19, 2025
  • xlstm Public

    Official repository of the xLSTM.

    NX-AI/xlstm’s past year of commit activity
    Python 1,992 Apache-2.0 156 45 5 Updated Sep 3, 2025
  • mlstm_kernels Public

    Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.

    NX-AI/mlstm_kernels’s past year of commit activity
    Jupyter Notebook 70 6 6 2 Updated Aug 14, 2025
  • flashrnn Public

    FlashRNN - Fast RNN Kernels with I/O Awareness

    NX-AI/flashrnn’s past year of commit activity
    Python 98 7 1 1 Updated Jun 12, 2025
  • RULER Public Forked from NVIDIA/RULER

    This repo contains the source code for RULER: What’s the Real Context Size of Your Long-Context Language Models?

    NX-AI/RULER’s past year of commit activity
    Python 0 Apache-2.0 114 0 0 Updated Feb 4, 2025
  • xlstm-jax Public

    Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.co/NX-AI/xLSTM-7b.

    NX-AI/xlstm-jax’s past year of commit activity
    Python 104 3 1 0 Updated Jan 8, 2025
  • vision-lstm Public

    xLSTM as Generic Vision Backbone

    NX-AI/vision-lstm’s past year of commit activity
    Python 488 Apache-2.0 35 2 0 Updated Nov 4, 2024