Source code, pretrained models and documentation for xLSTM and TiRex, released by NXAI GmbH, Linz, Austria. Each project links to its repository, paper and, where available, documentation and packages.
Pretrained multivariate time series forecasting model with past and future-known covariates and quantile forecasts, released for zero-shot use.
pip install tirex-2
Zero-shot univariate forecasting model for long and short horizons with enhanced in-context learning, based on xLSTM.
pip install tirex-ts
Official PyTorch implementation of the xLSTM architecture, including the xLSTM 7B language model.
pip install xlstm
Official JAX implementation of xLSTM with efficient training and inference code.
Vision-LSTM (ViL) adapts xLSTM as a generic backbone for computer vision. Includes pretrained models and a full training pipeline.
Code and data for the scaling behaviour of xLSTM and Transformer models from 80M to 7B parameters.
Tiled Flash Linear Attention: kernel library for fast and efficient mLSTM training and inference.
pip install mlstm_kernels
CUDA and Triton kernels for LSTM, GRU, Elman networks and sLSTM, optimised for modern GPU hardware.
pip install flashrnn
Triton kernels and a PyTorch layer for KronLSTM, a Kronecker delta rule for long-context linear attention.
Efficient normalisation via shared RMS statistics, reducing the number of operations on the critical path of each Transformer block.
| Project | GitHub stars | PyPI downloads | Hugging Face downloads |
|---|---|---|---|
| xLSTMNX-AI/xlstm | 2,214 | 366,93228,401 last 30 days | 181,8802,811 last 30 days |
| Vision-LSTMNX-AI/vision-lstm | 491 | – | – |
| TiRexNX-AI/tirex | 308 | 101,7299,263 last 30 days | 3,934,935141,037 last 30 days |
| FlashRNNNX-AI/flashrnn | 190 | 74,90030,266 last 30 days | – |
| TiRex-2NX-AI/tirex-2 | 161 | 48,74824,397 last 30 days | 23,2645,672 last 30 days |
| xLSTM JAXNX-AI/xlstm-jax | 105 | – | – |
| mLSTM KernelsNX-AI/mlstm_kernels | 96 | 262,29537,795 last 30 days | – |
| KronLSTMNX-AI/KronLSTM | 62 | – | 348348 last 30 days |
| xLSTM Scaling LawsNX-AI/xlstm_scaling_laws | 24 | – | 00 last 30 days |
| CoupledNormNX-AI/coupled-norm | 0 | – | – |