This specific version (1.1-gifteval) includes the 1.1 improvements plus the pretraining dataset has been cleaned
to remove overlaps with the GIFT-Eval test dataset.
TiRex is a
time-series foundation model designed for
time series forecasting,
with the emphasis to provide state-of-the-art forecasts for both short- and long-term forecasting horizon.
TiRex is
35M parameter small and is based on the
xLSTM architecture allowing fast and performant forecasts.
The model is described in the paper
TiRex: Zero-Shot Forecasting across Long and Short Horizons with Enhanced In-Context Learning.
-
Zero-Shot Forecasting:
TiRex performs forecasting without any training on your data. Just download and forecast.
-
Quantile Predictions:
TiRex not only provides point estimates but provides quantile estimates.
-
State-of-the-art Performance over Long and Short Horizons:
TiRex achieves top scores in various time series forecasting benchmarks, see
GiftEval and
ChronosZS.
These benchmark show that TiRex provides great performance for both long and short-term forecasting.
The inference code is available on
GitHub.
TiRex is currently only tested on
Linux systems and Nvidia GPUs with compute capability >= 8.0.
If you want to use different systems, please check the
FAQ in the code repository.
It's best to install TiRex in the specified conda environment.
The respective conda dependency file is
requirements_py26.yaml.
1# 1) Setup and activate conda env from ./requirements_py26.yaml
2git clone github.com/NX-AI/tirex
3conda env create --file ./tirex/requirements_py26.yaml
4conda activate tirex
5
6# 2) [Mandatory] Install Tirex
7
8## 2a) Install from source
9git clone github.com/NX-AI/tirex # if not already cloned before
10cd tirex
11pip install -e .
12
13# 2b) Install from PyPi (will be available soon)
14
15# 2) Optional: Install also optional dependencies
16pip install .[gluonts] # enable gluonTS in/output API
17pip install .[hfdataset] # enable HuggingFace datasets in/output API
18pip install .[notebooks] # To run the example notebooks
1import torch
2from tirex import load_model, ForecastModel
3
4model: ForecastModel = load_model("NX-AI/TiRex-1.1-gifteval")
5data = torch.rand((5, 128)) # Sample Data (5 time series with length 128)
6forecast = model.forecast(context=data, prediction_length=64, resample_strategy="frequency")
We provide an extended quick start example in the
GitHub repository.
If you have problems please check the FAQ / Troubleshooting section in the
GitHub repository
and feel free to create a GitHub issue or start a discussion.
1@article{auerTiRexZeroShotForecasting2025,
2 title = {{{TiRex}}: {{Zero-Shot Forecasting Across Long}} and {{Short Horizons}} with {{Enhanced In-Context Learning}}},
3 author = {Auer, Andreas and Podest, Patrick and Klotz, Daniel and B{\"o}ck, Sebastian and Klambauer, G{\"u}nter and Hochreiter, Sepp},
4 journal = {ArXiv},
5 volume = {2505.23719},
6 year = {2025}
7}
TiRex is licensed under the
NXAI community license.