├── .github ├── ISSUE_TEMPLATE │ ├── 01-feature_request.yml │ └── 02-bug_report.yml └── workflows │ ├── benchmark.yaml │ ├── cla.yaml │ ├── docker-build.yaml │ ├── docs.yaml │ ├── precommit.yaml │ ├── publish.yaml │ ├── sync.yaml │ ├── test-inference.yaml │ └── test.yaml ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── benchmark ├── Readme.md ├── benchmark.py ├── result │ ├── tirex-cuda-cuda-uncompiled-h100-2025-11-10.csv │ ├── tirex-torch-cpu-h100-2025-11-10.csv │ ├── tirex-torch-cpu-macbook-2025-11-10.csv │ ├── tirex-torch-cpu-uncompiled-h100-2025-11-10.csv │ ├── tirex-torch-cpu-uncompiled-macbook-2025-11-10.csv │ ├── tirex-torch-cuda-h100-2025-11-10.csv │ ├── tirex-torch-cuda-uncompiled-h100-2025-11-10.csv │ ├── tirex-torch-mps-macbook-2025-11-10.csv │ └── tirex-torch-mps-uncompiled-macbook-2025-11-10.csv └── visualize.ipynb ├── docs ├── README.md ├── content │ ├── benchmarks │ │ └── index.md │ ├── contributing │ │ └── index.md │ ├── deployment │ │ └── index.md │ ├── getting-started │ │ ├── install.md │ │ └── quickstart.md │ ├── how-to │ │ ├── anomaly.md │ │ ├── classification │ │ │ ├── index.md │ │ │ ├── practice.md │ │ │ ├── theory.md │ │ │ └── workflow.md │ │ ├── evaluation.md │ │ └── forecasting │ │ │ ├── index.md │ │ │ ├── practice.md │ │ │ ├── theory.md │ │ │ └── workflow.md │ ├── index.mdx │ └── introduction │ │ └── index.mdx ├── docusaurus.config.js ├── package-lock.json ├── package.json ├── scripts │ ├── generate_api_docs_sphinx.sh │ └── patch-gray-matter-js-yaml.js ├── sidebars.js ├── sphinx │ └── source │ │ ├── conf.py │ │ ├── index.rst │ │ ├── tirex.classification.rst │ │ └── tirex.rst ├── src │ ├── components │ │ └── Badges.tsx │ └── css │ │ └── custom.css └── static │ └── img │ ├── NXAI-Logo-white.png │ ├── NXAI-Logo.png │ ├── favicon.ico │ ├── long_horizon_prediction_l1024.png │ ├── long_horizon_prediction_l512.png │ ├── long_horizon_prediction_l768.png │ ├── short_horizon_future.png │ ├── short_horizon_prediction_l100.png │ ├── short_horizon_prediction_l12.png │ ├── short_horizon_prediction_l24.png │ ├── short_horizon_sample.png │ └── sine_wave.png ├── examples ├── chronos_zs │ └── chronos_zs.ipynb ├── gifteval │ ├── data_fixed_new_numpy.py │ ├── dataset_properties.json │ ├── gift_eval_utils.py │ └── gifteval.ipynb ├── quick_start_tirex.ipynb └── quick_start_tirex_classification.ipynb ├── inference ├── Dockerfile.cpu ├── Dockerfile.gpu ├── README.md ├── THIRD-PARTY-LICENSES.md ├── app │ ├── __init__.py │ ├── config.py │ ├── http_server.py │ ├── main.py │ ├── model.py │ └── mqtt_server.py ├── pytest.ini ├── requirements-dev.txt ├── requirements.txt └── tests │ ├── conftest.py │ ├── test_http.py │ ├── test_mcp.py │ └── test_mqtt.py ├── pyproject.toml ├── requirements_cu124.yaml ├── src └── tirex │ ├── __init__.py │ ├── api_adapter │ ├── __init__.py │ ├── forecast.py │ ├── gluon.py │ ├── hf_data.py │ └── standard_adapter.py │ ├── base.py │ ├── models │ ├── __init__.py │ ├── classification │ │ ├── __init__.py │ │ ├── embedding.py │ │ ├── heads │ │ │ ├── base_classifier.py │ │ │ ├── gbm_classifier.py │ │ │ ├── linear_classifier.py │ │ │ └── rf_classifier.py │ │ ├── trainer.py │ │ └── utils.py │ ├── patcher.py │ ├── slstm │ │ ├── block.py │ │ ├── cell.py │ │ └── layer.py │ └── tirex.py │ └── util.py └── tests ├── data ├── air_passengers.csv ├── air_passengers_forecast_ref.csv ├── air_passengers_quantiles_ref.pt ├── loop_seattle_5T.csv ├── loop_seattle_5T_forecast_ref.csv ├── loop_seattle_5T_forecast_resampled_ref.csv ├── loop_seattle_5T_quantiles_ref.pt └── loop_seattle_5T_quantiles_resampled_ref.pt ├── data_embeddings ├── context_data_wrapper.pt ├── embed_context_data.pt ├── embed_context_ref.pt ├── tirex_embedding_augmented_ref.pt └── tirex_embedding_ref.pt ├── test_chronos_zs.py ├── test_compile.py ├── test_embedding.py ├── test_forecast.py ├── test_forecast_adapter.py ├── test_gbm_classifier.py ├── test_linear_classifier.py ├── test_load_model.py ├── test_patcher.py ├── test_rf_classifier.py ├── test_slstm_torch_vs_cuda.py ├── test_standard_adapter.py └── test_util_freq.py /.github/ISSUE_TEMPLATE/01-feature_request.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.github/ISSUE_TEMPLATE/01-feature_request.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/02-bug_report.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.github/ISSUE_TEMPLATE/02-bug_report.yml -------------------------------------------------------------------------------- /.github/workflows/benchmark.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.github/workflows/benchmark.yaml -------------------------------------------------------------------------------- /.github/workflows/cla.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.github/workflows/cla.yaml -------------------------------------------------------------------------------- /.github/workflows/docker-build.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.github/workflows/docker-build.yaml -------------------------------------------------------------------------------- /.github/workflows/docs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.github/workflows/docs.yaml -------------------------------------------------------------------------------- /.github/workflows/precommit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.github/workflows/precommit.yaml -------------------------------------------------------------------------------- /.github/workflows/publish.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.github/workflows/publish.yaml -------------------------------------------------------------------------------- /.github/workflows/sync.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.github/workflows/sync.yaml -------------------------------------------------------------------------------- /.github/workflows/test-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.github/workflows/test-inference.yaml -------------------------------------------------------------------------------- /.github/workflows/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.github/workflows/test.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/README.md -------------------------------------------------------------------------------- /benchmark/Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/benchmark/Readme.md -------------------------------------------------------------------------------- /benchmark/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/benchmark/benchmark.py -------------------------------------------------------------------------------- /benchmark/result/tirex-cuda-cuda-uncompiled-h100-2025-11-10.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/benchmark/result/tirex-cuda-cuda-uncompiled-h100-2025-11-10.csv -------------------------------------------------------------------------------- /benchmark/result/tirex-torch-cpu-h100-2025-11-10.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/benchmark/result/tirex-torch-cpu-h100-2025-11-10.csv -------------------------------------------------------------------------------- /benchmark/result/tirex-torch-cpu-macbook-2025-11-10.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/benchmark/result/tirex-torch-cpu-macbook-2025-11-10.csv -------------------------------------------------------------------------------- /benchmark/result/tirex-torch-cpu-uncompiled-h100-2025-11-10.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/benchmark/result/tirex-torch-cpu-uncompiled-h100-2025-11-10.csv -------------------------------------------------------------------------------- /benchmark/result/tirex-torch-cpu-uncompiled-macbook-2025-11-10.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/benchmark/result/tirex-torch-cpu-uncompiled-macbook-2025-11-10.csv -------------------------------------------------------------------------------- /benchmark/result/tirex-torch-cuda-h100-2025-11-10.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/benchmark/result/tirex-torch-cuda-h100-2025-11-10.csv -------------------------------------------------------------------------------- /benchmark/result/tirex-torch-cuda-uncompiled-h100-2025-11-10.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/benchmark/result/tirex-torch-cuda-uncompiled-h100-2025-11-10.csv -------------------------------------------------------------------------------- /benchmark/result/tirex-torch-mps-macbook-2025-11-10.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/benchmark/result/tirex-torch-mps-macbook-2025-11-10.csv -------------------------------------------------------------------------------- /benchmark/result/tirex-torch-mps-uncompiled-macbook-2025-11-10.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/benchmark/result/tirex-torch-mps-uncompiled-macbook-2025-11-10.csv -------------------------------------------------------------------------------- /benchmark/visualize.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/benchmark/visualize.ipynb -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/content/benchmarks/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/benchmarks/index.md -------------------------------------------------------------------------------- /docs/content/contributing/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/contributing/index.md -------------------------------------------------------------------------------- /docs/content/deployment/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/deployment/index.md -------------------------------------------------------------------------------- /docs/content/getting-started/install.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/getting-started/install.md -------------------------------------------------------------------------------- /docs/content/getting-started/quickstart.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/getting-started/quickstart.md -------------------------------------------------------------------------------- /docs/content/how-to/anomaly.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/how-to/anomaly.md -------------------------------------------------------------------------------- /docs/content/how-to/classification/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/how-to/classification/index.md -------------------------------------------------------------------------------- /docs/content/how-to/classification/practice.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/how-to/classification/practice.md -------------------------------------------------------------------------------- /docs/content/how-to/classification/theory.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/how-to/classification/theory.md -------------------------------------------------------------------------------- /docs/content/how-to/classification/workflow.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/how-to/classification/workflow.md -------------------------------------------------------------------------------- /docs/content/how-to/evaluation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/how-to/evaluation.md -------------------------------------------------------------------------------- /docs/content/how-to/forecasting/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/how-to/forecasting/index.md -------------------------------------------------------------------------------- /docs/content/how-to/forecasting/practice.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/how-to/forecasting/practice.md -------------------------------------------------------------------------------- /docs/content/how-to/forecasting/theory.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/how-to/forecasting/theory.md -------------------------------------------------------------------------------- /docs/content/how-to/forecasting/workflow.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/how-to/forecasting/workflow.md -------------------------------------------------------------------------------- /docs/content/index.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/index.mdx -------------------------------------------------------------------------------- /docs/content/introduction/index.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/content/introduction/index.mdx -------------------------------------------------------------------------------- /docs/docusaurus.config.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/docusaurus.config.js -------------------------------------------------------------------------------- /docs/package-lock.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/package-lock.json -------------------------------------------------------------------------------- /docs/package.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/package.json -------------------------------------------------------------------------------- /docs/scripts/generate_api_docs_sphinx.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/scripts/generate_api_docs_sphinx.sh -------------------------------------------------------------------------------- /docs/scripts/patch-gray-matter-js-yaml.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/scripts/patch-gray-matter-js-yaml.js -------------------------------------------------------------------------------- /docs/sidebars.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/sidebars.js -------------------------------------------------------------------------------- /docs/sphinx/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/sphinx/source/conf.py -------------------------------------------------------------------------------- /docs/sphinx/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/sphinx/source/index.rst -------------------------------------------------------------------------------- /docs/sphinx/source/tirex.classification.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/sphinx/source/tirex.classification.rst -------------------------------------------------------------------------------- /docs/sphinx/source/tirex.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/sphinx/source/tirex.rst -------------------------------------------------------------------------------- /docs/src/components/Badges.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/src/components/Badges.tsx -------------------------------------------------------------------------------- /docs/src/css/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/src/css/custom.css -------------------------------------------------------------------------------- /docs/static/img/NXAI-Logo-white.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/static/img/NXAI-Logo-white.png -------------------------------------------------------------------------------- /docs/static/img/NXAI-Logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/static/img/NXAI-Logo.png -------------------------------------------------------------------------------- /docs/static/img/favicon.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/static/img/favicon.ico -------------------------------------------------------------------------------- /docs/static/img/long_horizon_prediction_l1024.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/static/img/long_horizon_prediction_l1024.png -------------------------------------------------------------------------------- /docs/static/img/long_horizon_prediction_l512.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/static/img/long_horizon_prediction_l512.png -------------------------------------------------------------------------------- /docs/static/img/long_horizon_prediction_l768.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/static/img/long_horizon_prediction_l768.png -------------------------------------------------------------------------------- /docs/static/img/short_horizon_future.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/static/img/short_horizon_future.png -------------------------------------------------------------------------------- /docs/static/img/short_horizon_prediction_l100.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/static/img/short_horizon_prediction_l100.png -------------------------------------------------------------------------------- /docs/static/img/short_horizon_prediction_l12.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/static/img/short_horizon_prediction_l12.png -------------------------------------------------------------------------------- /docs/static/img/short_horizon_prediction_l24.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/static/img/short_horizon_prediction_l24.png -------------------------------------------------------------------------------- /docs/static/img/short_horizon_sample.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/static/img/short_horizon_sample.png -------------------------------------------------------------------------------- /docs/static/img/sine_wave.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/docs/static/img/sine_wave.png -------------------------------------------------------------------------------- /examples/chronos_zs/chronos_zs.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/examples/chronos_zs/chronos_zs.ipynb -------------------------------------------------------------------------------- /examples/gifteval/data_fixed_new_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/examples/gifteval/data_fixed_new_numpy.py -------------------------------------------------------------------------------- /examples/gifteval/dataset_properties.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/examples/gifteval/dataset_properties.json -------------------------------------------------------------------------------- /examples/gifteval/gift_eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/examples/gifteval/gift_eval_utils.py -------------------------------------------------------------------------------- /examples/gifteval/gifteval.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/examples/gifteval/gifteval.ipynb -------------------------------------------------------------------------------- /examples/quick_start_tirex.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/examples/quick_start_tirex.ipynb -------------------------------------------------------------------------------- /examples/quick_start_tirex_classification.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/examples/quick_start_tirex_classification.ipynb -------------------------------------------------------------------------------- /inference/Dockerfile.cpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/Dockerfile.cpu -------------------------------------------------------------------------------- /inference/Dockerfile.gpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/Dockerfile.gpu -------------------------------------------------------------------------------- /inference/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/README.md -------------------------------------------------------------------------------- /inference/THIRD-PARTY-LICENSES.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/THIRD-PARTY-LICENSES.md -------------------------------------------------------------------------------- /inference/app/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/app/__init__.py -------------------------------------------------------------------------------- /inference/app/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/app/config.py -------------------------------------------------------------------------------- /inference/app/http_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/app/http_server.py -------------------------------------------------------------------------------- /inference/app/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/app/main.py -------------------------------------------------------------------------------- /inference/app/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/app/model.py -------------------------------------------------------------------------------- /inference/app/mqtt_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/app/mqtt_server.py -------------------------------------------------------------------------------- /inference/pytest.ini: -------------------------------------------------------------------------------- 1 | [pytest] 2 | asyncio_mode = auto 3 | -------------------------------------------------------------------------------- /inference/requirements-dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/requirements-dev.txt -------------------------------------------------------------------------------- /inference/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/requirements.txt -------------------------------------------------------------------------------- /inference/tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/tests/conftest.py -------------------------------------------------------------------------------- /inference/tests/test_http.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/tests/test_http.py -------------------------------------------------------------------------------- /inference/tests/test_mcp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/tests/test_mcp.py -------------------------------------------------------------------------------- /inference/tests/test_mqtt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/inference/tests/test_mqtt.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements_cu124.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/requirements_cu124.yaml -------------------------------------------------------------------------------- /src/tirex/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/__init__.py -------------------------------------------------------------------------------- /src/tirex/api_adapter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/api_adapter/__init__.py -------------------------------------------------------------------------------- /src/tirex/api_adapter/forecast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/api_adapter/forecast.py -------------------------------------------------------------------------------- /src/tirex/api_adapter/gluon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/api_adapter/gluon.py -------------------------------------------------------------------------------- /src/tirex/api_adapter/hf_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/api_adapter/hf_data.py -------------------------------------------------------------------------------- /src/tirex/api_adapter/standard_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/api_adapter/standard_adapter.py -------------------------------------------------------------------------------- /src/tirex/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/base.py -------------------------------------------------------------------------------- /src/tirex/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/__init__.py -------------------------------------------------------------------------------- /src/tirex/models/classification/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/classification/__init__.py -------------------------------------------------------------------------------- /src/tirex/models/classification/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/classification/embedding.py -------------------------------------------------------------------------------- /src/tirex/models/classification/heads/base_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/classification/heads/base_classifier.py -------------------------------------------------------------------------------- /src/tirex/models/classification/heads/gbm_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/classification/heads/gbm_classifier.py -------------------------------------------------------------------------------- /src/tirex/models/classification/heads/linear_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/classification/heads/linear_classifier.py -------------------------------------------------------------------------------- /src/tirex/models/classification/heads/rf_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/classification/heads/rf_classifier.py -------------------------------------------------------------------------------- /src/tirex/models/classification/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/classification/trainer.py -------------------------------------------------------------------------------- /src/tirex/models/classification/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/classification/utils.py -------------------------------------------------------------------------------- /src/tirex/models/patcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/patcher.py -------------------------------------------------------------------------------- /src/tirex/models/slstm/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/slstm/block.py -------------------------------------------------------------------------------- /src/tirex/models/slstm/cell.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/slstm/cell.py -------------------------------------------------------------------------------- /src/tirex/models/slstm/layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/slstm/layer.py -------------------------------------------------------------------------------- /src/tirex/models/tirex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/models/tirex.py -------------------------------------------------------------------------------- /src/tirex/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/src/tirex/util.py -------------------------------------------------------------------------------- /tests/data/air_passengers.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data/air_passengers.csv -------------------------------------------------------------------------------- /tests/data/air_passengers_forecast_ref.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data/air_passengers_forecast_ref.csv -------------------------------------------------------------------------------- /tests/data/air_passengers_quantiles_ref.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data/air_passengers_quantiles_ref.pt -------------------------------------------------------------------------------- /tests/data/loop_seattle_5T.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data/loop_seattle_5T.csv -------------------------------------------------------------------------------- /tests/data/loop_seattle_5T_forecast_ref.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data/loop_seattle_5T_forecast_ref.csv -------------------------------------------------------------------------------- /tests/data/loop_seattle_5T_forecast_resampled_ref.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data/loop_seattle_5T_forecast_resampled_ref.csv -------------------------------------------------------------------------------- /tests/data/loop_seattle_5T_quantiles_ref.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data/loop_seattle_5T_quantiles_ref.pt -------------------------------------------------------------------------------- /tests/data/loop_seattle_5T_quantiles_resampled_ref.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data/loop_seattle_5T_quantiles_resampled_ref.pt -------------------------------------------------------------------------------- /tests/data_embeddings/context_data_wrapper.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data_embeddings/context_data_wrapper.pt -------------------------------------------------------------------------------- /tests/data_embeddings/embed_context_data.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data_embeddings/embed_context_data.pt -------------------------------------------------------------------------------- /tests/data_embeddings/embed_context_ref.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data_embeddings/embed_context_ref.pt -------------------------------------------------------------------------------- /tests/data_embeddings/tirex_embedding_augmented_ref.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data_embeddings/tirex_embedding_augmented_ref.pt -------------------------------------------------------------------------------- /tests/data_embeddings/tirex_embedding_ref.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/data_embeddings/tirex_embedding_ref.pt -------------------------------------------------------------------------------- /tests/test_chronos_zs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_chronos_zs.py -------------------------------------------------------------------------------- /tests/test_compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_compile.py -------------------------------------------------------------------------------- /tests/test_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_embedding.py -------------------------------------------------------------------------------- /tests/test_forecast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_forecast.py -------------------------------------------------------------------------------- /tests/test_forecast_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_forecast_adapter.py -------------------------------------------------------------------------------- /tests/test_gbm_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_gbm_classifier.py -------------------------------------------------------------------------------- /tests/test_linear_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_linear_classifier.py -------------------------------------------------------------------------------- /tests/test_load_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_load_model.py -------------------------------------------------------------------------------- /tests/test_patcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_patcher.py -------------------------------------------------------------------------------- /tests/test_rf_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_rf_classifier.py -------------------------------------------------------------------------------- /tests/test_slstm_torch_vs_cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_slstm_torch_vs_cuda.py -------------------------------------------------------------------------------- /tests/test_standard_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_standard_adapter.py -------------------------------------------------------------------------------- /tests/test_util_freq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NX-AI/tirex/HEAD/tests/test_util_freq.py --------------------------------------------------------------------------------