tirex-mirror 2025.6.9__tar.gz

tirex_mirror-2025.6.9/LICENSE

@@ -0,0 +1,49 @@
+NXAI COMMUNITY LICENSE AGREEMENT
+
+Preamble 1
+
+We are proud to present the NXAI TiRex time series model and software, demonstrating the strength of xLSTM for time series. While TiRex is freely available for open research and development, we believe that organizations significantly benefiting from our technology should contribute back. Our goal is to support research, small and medium-sized enterprises (SMEs), and open innovation, while ensuring that large enterprises who incorporate TiRex into commercial products or services fairly compensate the creators for their research and development efforts.
+
+Linz, May 27, 2025.
+
+Preamble 2
+
+The NXAI COMMUNITY LICENSE AGREEMENT is based on the META LLAMA 3 COMMUNITY LICENSE AGREEMENT and contains some modifications, especially Section 2, “Additional Commercial Terms” is different.
+
+“Agreement” means the terms and conditions for use, reproduction, distribution and modification of the NXAI Materials set forth herein.
+“Documentation” means the specifications, manuals and documentation accompanying NXAI Materials distributed by NXAI at https://github.com/NX-AI/.
+“Licensee” or “you” means you, or your employer or any other person or entity (if you are entering into this Agreement on such person or entity’s behalf), of the age required under applicable laws, rules or regulations to provide legal consent and that has legal authority to bind your employer or such other person or entity if you are entering in this Agreement on their behalf.
+“NXAI Materials” means, collectively, NXAI’s proprietary models, algorithms and any Software, including machine-learning model code, trained model weights, inference-enabling code, training-enabling code, fine-tuning enabling code and all other work of NXAI in the field of neural networks, Documentation (and any portion thereof) made available under this Agreement.
+“NXAI” or “we” means NXAI GmbH, Linz, Austria. Contact: license@nx-ai.com
+
+By using or distributing any portion or element of the NXAI Materials, you agree to be bound by this Agreement.
+
+1. License Rights and Redistribution.
+
+    a. Grant of Rights. You are granted a non-exclusive, worldwide, non-transferable and royalty-free limited license under NXAIs intellectual property embodied in the NXAI Materials to use, reproduce, distribute, copy, create derivative works of, and make modifications to the NXAI Materials.
+
+    b. Redistribution and Use.
+
+        i. If you distribute or make available the NXAI Materials (or any derivative works thereof), or a product or service that uses any of them, including another AI model, you shall (A) provide a copy of this Agreement with any such NXAI Materials; and (B) prominently display “Built with technology from NXAI” on a related website, user interface, blogpost, about page, or product documentation.
+
+        ii. If you receive NXAI Materials, or any derivative works thereof, from a Licensee as part of an integrated end user product, then Section 2 of this Agreement will not apply to you.
+
+        iii. You must retain in all copies of the NXAI Materials that you distribute the following attribution notice within a “Notice” text file distributed as a part of such copies: “This product includes materials developed at NXAI that are licensed under the NXAI Community License, Copyright © NXAI GmbH, All Rights Reserved.”
+
+2. Additional Commercial Terms. If (a) the Licensee, on a consolidated basis (including parent, subsidiaries, and affiliates), exceeds the annual revenue of one hundred million Euros (€100,000,000), and (b) the Licensee incorporates NXAI Material, in whole or in part, into a Commercial Product or Service, then the Licensee must obtain a commercial license from NXAI, which NXAI may grant to you in its sole discretion, and you are not authorized to exercise any of the rights under this Agreement unless or until NXAI otherwise expressly grants you such rights.
+
+3. Disclaimer of Warranty. UNLESS REQUIRED BY APPLICABLE LAW, THE NXAI MATERIALS AND ANY OUTPUT AND RESULTS THEREFROM ARE PROVIDED ON AN “AS IS” BASIS, WITHOUT WARRANTIES OF ANY KIND, AND NXAI DISCLAIMS ALL WARRANTIES OF ANY KIND, BOTH EXPRESS AND IMPLIED, INCLUDING, WITHOUT LIMITATION, ANY WARRANTIES OF TITLE, NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE. YOU ARE SOLELY RESPONSIBLE FOR DETERMINING THE APPROPRIATENESS OF USING OR REDISTRIBUTING THE NXAI MATERIALS AND ASSUME ANY RISKS ASSOCIATED WITH YOUR USE OF THE NXAI MATERIALS AND ANY OUTPUT AND RESULTS.
+
+4. Limitation of Liability. IN NO EVENT WILL NXAI OR ITS AFFILIATES BE LIABLE UNDER ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, TORT, NEGLIGENCE, PRODUCTS LIABILITY, OR OTHERWISE, ARISING OUT OF THIS AGREEMENT, FOR ANY LOST PROFITS OR ANY INDIRECT, SPECIAL, CONSEQUENTIAL, INCIDENTAL, EXEMPLARY OR PUNITIVE DAMAGES, EVEN IF NXAI OR ITS AFFILIATES HAVE BEEN ADVISED OF THE POSSIBILITY OF ANY OF THE FOREGOING.
+
+5. Intellectual Property.
+
+    a. No trademark licenses are granted under this Agreement, and in connection with the NXAI Materials, neither NXAI nor Licensee may use any name or mark owned by or associated with the other or any of its affiliates, except as required for reasonable and customary use in describing and redistributing the NXAI Materials or as set forth in this Section 5(a). NXAI hereby grants you a license to use “NXAI” (the “Mark”) solely as required to comply with the last sentence of Section 1.b.i. All goodwill arising out of your use of the Mark will insure to the benefit of NXAI.
+
+    b. Subject to NXAI’s ownership of NXAI Materials and derivatives made by or for NXAI, with respect to any derivative works and modifications of the NXAI Materials that are made by you, as between you and NXAI, you are and will be the owner of such derivative works and modifications.
+
+c. If you institute litigation or other proceedings against NXAI or any entity (including a cross-claim or counterclaim in a lawsuit) alleging that the NXAI Materials or models released by NXAI outputs or results, or any portion of any of the foregoing, constitutes infringement of intellectual property or other rights owned or licensable by you, then any licenses granted to you under this Agreement shall terminate as of the date such litigation or claim is filed or instituted. You will indemnify and hold harmless NXAI from and against any claim by any third party arising out of or related to your use or distribution of the NXAI Materials.
+
+6. Term and Termination. The term of this Agreement will commence upon your acceptance of this Agreement or access to the NXAI Materials and will continue in full force and effect until terminated in accordance with the terms and conditions herein. NXAI may terminate this Agreement if you are in breach of any term or condition of this Agreement. Upon termination of this Agreement, you shall delete and cease use of the NXAI Materials. Sections 3, 4 and 7 shall survive the termination of this Agreement.
+
+7. Governing Law and Jurisdiction. This Agreement shall be governed by and construed in accordance with the laws of the Republic of Austria, without regard to its conflict of laws principles. The courts located in Linz, Austria shall have exclusive jurisdiction over any disputes arising out of or in connection with this Agreement.

tirex_mirror-2025.6.9/LICENSE_MIRROR.txt

@@ -0,0 +1,7 @@
+This project, `tirex-mirror`, is an unofficial mirror and repackaging of the TiRex model developed by NXAI GmbH.
+
+It is maintained solely to enable installation via PyPI and does not modify or extend the original code or functionality.
+
+This repository is not affiliated with or endorsed by NXAI GmbH.
+
+All rights to the original code remain with NXAI GmbH. Please refer to `LICENSE.txt` and `NOTICE.txt` for licensing terms and attribution requirements as defined by the NXAI Community License.

tirex_mirror-2025.6.9/MANIFEST.in

@@ -0,0 +1,3 @@
+ include LICENSE
+include NOTICE.txt
+include LICENSE_MIRROR.txt

tirex_mirror-2025.6.9/NOTICE.txt

@@ -0,0 +1 @@
+This product includes materials developed at NXAI that are licensed under the NXAI Community License, Copyright © NXAI GmbH, All Rights Reserved.

tirex_mirror-2025.6.9/PKG-INFO

@@ -0,0 +1,180 @@
+Metadata-Version: 2.4
+Name: tirex-mirror
+Version: 2025.6.9
+Summary: Unofficial mirror of NX-AI/tirex for packaging
+Author-email: Arpad Rozsas <rozsasarpi@gmail.com>
+License: NXAI COMMUNITY LICENSE AGREEMENT
+        
+        Preamble 1
+        
+        We are proud to present the NXAI TiRex time series model and software, demonstrating the strength of xLSTM for time series. While TiRex is freely available for open research and development, we believe that organizations significantly benefiting from our technology should contribute back. Our goal is to support research, small and medium-sized enterprises (SMEs), and open innovation, while ensuring that large enterprises who incorporate TiRex into commercial products or services fairly compensate the creators for their research and development efforts.
+        
+        Linz, May 27, 2025.
+        
+        Preamble 2
+        
+        The NXAI COMMUNITY LICENSE AGREEMENT is based on the META LLAMA 3 COMMUNITY LICENSE AGREEMENT and contains some modifications, especially Section 2, “Additional Commercial Terms” is different.
+        
+        “Agreement” means the terms and conditions for use, reproduction, distribution and modification of the NXAI Materials set forth herein.
+        “Documentation” means the specifications, manuals and documentation accompanying NXAI Materials distributed by NXAI at https://github.com/NX-AI/.
+        “Licensee” or “you” means you, or your employer or any other person or entity (if you are entering into this Agreement on such person or entity’s behalf), of the age required under applicable laws, rules or regulations to provide legal consent and that has legal authority to bind your employer or such other person or entity if you are entering in this Agreement on their behalf.
+        “NXAI Materials” means, collectively, NXAI’s proprietary models, algorithms and any Software, including machine-learning model code, trained model weights, inference-enabling code, training-enabling code, fine-tuning enabling code and all other work of NXAI in the field of neural networks, Documentation (and any portion thereof) made available under this Agreement.
+        “NXAI” or “we” means NXAI GmbH, Linz, Austria. Contact: license@nx-ai.com
+        
+        By using or distributing any portion or element of the NXAI Materials, you agree to be bound by this Agreement.
+        
+        1. License Rights and Redistribution.
+        
+            a. Grant of Rights. You are granted a non-exclusive, worldwide, non-transferable and royalty-free limited license under NXAIs intellectual property embodied in the NXAI Materials to use, reproduce, distribute, copy, create derivative works of, and make modifications to the NXAI Materials.
+        
+            b. Redistribution and Use.
+        
+                i. If you distribute or make available the NXAI Materials (or any derivative works thereof), or a product or service that uses any of them, including another AI model, you shall (A) provide a copy of this Agreement with any such NXAI Materials; and (B) prominently display “Built with technology from NXAI” on a related website, user interface, blogpost, about page, or product documentation.
+        
+                ii. If you receive NXAI Materials, or any derivative works thereof, from a Licensee as part of an integrated end user product, then Section 2 of this Agreement will not apply to you.
+        
+                iii. You must retain in all copies of the NXAI Materials that you distribute the following attribution notice within a “Notice” text file distributed as a part of such copies: “This product includes materials developed at NXAI that are licensed under the NXAI Community License, Copyright © NXAI GmbH, All Rights Reserved.”
+        
+        2. Additional Commercial Terms. If (a) the Licensee, on a consolidated basis (including parent, subsidiaries, and affiliates), exceeds the annual revenue of one hundred million Euros (€100,000,000), and (b) the Licensee incorporates NXAI Material, in whole or in part, into a Commercial Product or Service, then the Licensee must obtain a commercial license from NXAI, which NXAI may grant to you in its sole discretion, and you are not authorized to exercise any of the rights under this Agreement unless or until NXAI otherwise expressly grants you such rights.
+        
+        3. Disclaimer of Warranty. UNLESS REQUIRED BY APPLICABLE LAW, THE NXAI MATERIALS AND ANY OUTPUT AND RESULTS THEREFROM ARE PROVIDED ON AN “AS IS” BASIS, WITHOUT WARRANTIES OF ANY KIND, AND NXAI DISCLAIMS ALL WARRANTIES OF ANY KIND, BOTH EXPRESS AND IMPLIED, INCLUDING, WITHOUT LIMITATION, ANY WARRANTIES OF TITLE, NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE. YOU ARE SOLELY RESPONSIBLE FOR DETERMINING THE APPROPRIATENESS OF USING OR REDISTRIBUTING THE NXAI MATERIALS AND ASSUME ANY RISKS ASSOCIATED WITH YOUR USE OF THE NXAI MATERIALS AND ANY OUTPUT AND RESULTS.
+        
+        4. Limitation of Liability. IN NO EVENT WILL NXAI OR ITS AFFILIATES BE LIABLE UNDER ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, TORT, NEGLIGENCE, PRODUCTS LIABILITY, OR OTHERWISE, ARISING OUT OF THIS AGREEMENT, FOR ANY LOST PROFITS OR ANY INDIRECT, SPECIAL, CONSEQUENTIAL, INCIDENTAL, EXEMPLARY OR PUNITIVE DAMAGES, EVEN IF NXAI OR ITS AFFILIATES HAVE BEEN ADVISED OF THE POSSIBILITY OF ANY OF THE FOREGOING.
+        
+        5. Intellectual Property.
+        
+            a. No trademark licenses are granted under this Agreement, and in connection with the NXAI Materials, neither NXAI nor Licensee may use any name or mark owned by or associated with the other or any of its affiliates, except as required for reasonable and customary use in describing and redistributing the NXAI Materials or as set forth in this Section 5(a). NXAI hereby grants you a license to use “NXAI” (the “Mark”) solely as required to comply with the last sentence of Section 1.b.i. All goodwill arising out of your use of the Mark will insure to the benefit of NXAI.
+        
+            b. Subject to NXAI’s ownership of NXAI Materials and derivatives made by or for NXAI, with respect to any derivative works and modifications of the NXAI Materials that are made by you, as between you and NXAI, you are and will be the owner of such derivative works and modifications.
+        
+        c. If you institute litigation or other proceedings against NXAI or any entity (including a cross-claim or counterclaim in a lawsuit) alleging that the NXAI Materials or models released by NXAI outputs or results, or any portion of any of the foregoing, constitutes infringement of intellectual property or other rights owned or licensable by you, then any licenses granted to you under this Agreement shall terminate as of the date such litigation or claim is filed or instituted. You will indemnify and hold harmless NXAI from and against any claim by any third party arising out of or related to your use or distribution of the NXAI Materials.
+        
+        6. Term and Termination. The term of this Agreement will commence upon your acceptance of this Agreement or access to the NXAI Materials and will continue in full force and effect until terminated in accordance with the terms and conditions herein. NXAI may terminate this Agreement if you are in breach of any term or condition of this Agreement. Upon termination of this Agreement, you shall delete and cease use of the NXAI Materials. Sections 3, 4 and 7 shall survive the termination of this Agreement.
+        
+        7. Governing Law and Jurisdiction. This Agreement shall be governed by and construed in accordance with the laws of the Republic of Austria, without regard to its conflict of laws principles. The courts located in Linz, Austria shall have exclusive jurisdiction over any disputes arising out of or in connection with this Agreement.
+        
+Project-URL: Repository, https://github.com/rozsasarpi/tirex-mirror
+Project-URL: Issues, https://github.com/rozsasarpi/tirex-mirror/issues
+Keywords: TiRex,xLSTM,Time Series,Zero-shot,Deep Learning
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+License-File: LICENSE_MIRROR.txt
+License-File: NOTICE.txt
+Requires-Dist: torch
+Requires-Dist: torchvision
+Requires-Dist: xlstm
+Requires-Dist: einops
+Requires-Dist: ninja
+Requires-Dist: huggingface-hub
+Requires-Dist: lightning
+Requires-Dist: numpy
+Requires-Dist: pandas
+Requires-Dist: dacite
+Requires-Dist: tqdm
+Provides-Extra: notebooks
+Requires-Dist: ipykernel; extra == "notebooks"
+Provides-Extra: gluonts
+Requires-Dist: gluonts; extra == "gluonts"
+Provides-Extra: hfdataset
+Requires-Dist: datasets; extra == "hfdataset"
+Provides-Extra: all
+Requires-Dist: ipykernel; extra == "all"
+Requires-Dist: gluonts; extra == "all"
+Requires-Dist: datasets; extra == "all"
+Dynamic: license-file
+
+# tirex-mirror
+
+Unofficial mirror of [NX-AI/tirex](https://github.com/NX-AI/tirex) for packaging.
+
+## Legal Notice
+
+This is an unofficial mirror of [NX-AI/tirex](https://github.com/NX-AI/tirex).
+See [NOTICE.txt](NOTICE.txt) and [LICENSE_MIRROR.txt](LICENSE_MIRROR.txt) for attribution details.
+The original project is licensed under the terms in [LICENSE](LICENSE).
+
+## Purpose
+
+This repository automatically mirrors the upstream TiRex project and publishes it to PyPI as `tirex-mirror` for packaging and distribution purposes.
+
+**Installation**: `pip install tirex-mirror`  
+**Import**: `import tirex`
+
+## Development Notes
+
+### Setting up local environment
+
+```bash
+# Create and activate virtual environment
+python -m venv venv
+source venv/bin/activate
+
+# Install build dependencies
+pip install build toml twine
+```
+
+### Dry run workflow steps locally
+
+Run commands from `publish.yml`.
+
+Test building the package
+```bash
+python -m build
+```
+
+### Manual publishing
+
+Check package contents
+```bash
+twine check dist/*
+```
+
+Test upload to TestPyPI (optional)
+```bash
+twine upload --repository testpypi dist/*
+```
+
+Real upload to PyPI
+```bash
+twine upload dist/*
+```
+
+Test the package (from testpypi):
+```bash
+pip install \
+  --index-url https://test.pypi.org/simple/ \
+  --extra-index-url https://pypi.org/simple \
+  tirex-mirror==2025.06.09
+```
+
+### GitHub Releases
+
+The workflow automatically creates GitHub releases alongside PyPI releases.
+
+```bash
+# View releases
+gh release list
+
+# Create manual release (if needed)
+VERSION="2025.06.09"
+gh release create "v$VERSION" \
+  --title "TiRex Mirror v$VERSION" \
+  --notes "Manual release" \
+  dist/*
+
+# Manual workflow trigger
+gh workflow run publish.yml
+
+# Check workflow status
+gh run list --workflow=publish.yml
+```
+
+### Cleanup after testing
+
+Remove all untracked and ignored files and folders
+```bash
+git ls-files -o -i --exclude-standard | grep -vE '^(\.venv|venv)/' | xargs -r rm -rf
+```

tirex_mirror-2025.6.9/README.md

@@ -0,0 +1,92 @@
+# tirex-mirror
+
+Unofficial mirror of [NX-AI/tirex](https://github.com/NX-AI/tirex) for packaging.
+
+## Legal Notice
+
+This is an unofficial mirror of [NX-AI/tirex](https://github.com/NX-AI/tirex).
+See [NOTICE.txt](NOTICE.txt) and [LICENSE_MIRROR.txt](LICENSE_MIRROR.txt) for attribution details.
+The original project is licensed under the terms in [LICENSE](LICENSE).
+
+## Purpose
+
+This repository automatically mirrors the upstream TiRex project and publishes it to PyPI as `tirex-mirror` for packaging and distribution purposes.
+
+**Installation**: `pip install tirex-mirror`  
+**Import**: `import tirex`
+
+## Development Notes
+
+### Setting up local environment
+
+```bash
+# Create and activate virtual environment
+python -m venv venv
+source venv/bin/activate
+
+# Install build dependencies
+pip install build toml twine
+```
+
+### Dry run workflow steps locally
+
+Run commands from `publish.yml`.
+
+Test building the package
+```bash
+python -m build
+```
+
+### Manual publishing
+
+Check package contents
+```bash
+twine check dist/*
+```
+
+Test upload to TestPyPI (optional)
+```bash
+twine upload --repository testpypi dist/*
+```
+
+Real upload to PyPI
+```bash
+twine upload dist/*
+```
+
+Test the package (from testpypi):
+```bash
+pip install \
+  --index-url https://test.pypi.org/simple/ \
+  --extra-index-url https://pypi.org/simple \
+  tirex-mirror==2025.06.09
+```
+
+### GitHub Releases
+
+The workflow automatically creates GitHub releases alongside PyPI releases.
+
+```bash
+# View releases
+gh release list
+
+# Create manual release (if needed)
+VERSION="2025.06.09"
+gh release create "v$VERSION" \
+  --title "TiRex Mirror v$VERSION" \
+  --notes "Manual release" \
+  dist/*
+
+# Manual workflow trigger
+gh workflow run publish.yml
+
+# Check workflow status
+gh run list --workflow=publish.yml
+```
+
+### Cleanup after testing
+
+Remove all untracked and ignored files and folders
+```bash
+git ls-files -o -i --exclude-standard | grep -vE '^(\.venv|venv)/' | xargs -r rm -rf
+```

tirex_mirror-2025.6.9/pyproject.toml

@@ -0,0 +1,60 @@
+[project]
+name = "tirex-mirror"
+version = "2025.06.09"
+description = "Unofficial mirror of NX-AI/tirex for packaging"
+readme = "README.md"
+requires-python = ">=3.11"
+classifiers = [ "Programming Language :: Python :: 3", "Operating System :: OS Independent",]
+keywords = [ "TiRex", "xLSTM", "Time Series", "Zero-shot", "Deep Learning",]
+dependencies = [ "torch", "torchvision", "xlstm", "einops", "ninja", "huggingface-hub", "lightning", "numpy", "pandas", "dacite", "tqdm",]
+[[project.authors]]
+name = "Arpad Rozsas"
+email = "rozsasarpi@gmail.com"
+
+[build-system]
+requires = [ "setuptools>=42", "wheel",]
+build-backend = "setuptools.build_meta"
+
+[project.license]
+file = "LICENSE"
+
+[project.urls]
+Repository = "https://github.com/rozsasarpi/tirex-mirror"
+Issues = "https://github.com/rozsasarpi/tirex-mirror/issues"
+
+[project.optional-dependencies]
+notebooks = [ "ipykernel",]
+gluonts = [ "gluonts",]
+hfdataset = [ "datasets",]
+all = [ "ipykernel", "gluonts", "datasets",]
+
+[tool.docformatter]
+diff = false
+docstring-length = [ 40, 1000,]
+in-place = true
+pre-summary-newline = true
+recursive = true
+wrap-summaries = 100
+wrap-descriptions = 100
+non-cap = [ "TiRex", "xLSTM", "sLSTM",]
+
+[tool.ruff]
+exclude = [ ".eggs", ".git", ".ipynb_checkpoints", ".pytest_cache", ".ruff_cache", ".vscode", "__pypackages__", "_build", "build", "dist", "node_modules", "site-packages", "archive",]
+line-length = 120
+target-version = "py311"
+
+[tool.ruff.format]
+quote-style = "double"
+
+[tool.ruff.lint]
+select = [ "I001",]
+
+[tool.ruff.lint.mccabe]
+max-complexity = 18
+
+[tool.ruff.lint.isort]
+combine-as-imports = true
+section-order = [ "future", "standard-library", "third-party", "first-party", "tirex", "local-folder",]
+
+[tool.ruff.lint.isort.sections]
+tirex = [ "tirex",]

tirex_mirror-2025.6.9/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

tirex_mirror-2025.6.9/src/tirex/__init__.py

@@ -0,0 +1,8 @@
+# Copyright (c) NXAI GmbH.
+# This software may be used and distributed according to the terms of the NXAI Community License Agreement.
+
+from .api_adapter.forecast import ForecastModel
+from .base import load_model
+from .models.tirex import TiRexZero
+
+__all__ = ["load_model", "ForecastModel"]

tirex_mirror-2025.6.9/src/tirex/api_adapter/__init__.py

@@ -0,0 +1,2 @@
+# Copyright (c) NXAI GmbH.
+# This software may be used and distributed according to the terms of the NXAI Community License Agreement.

tirex_mirror-2025.6.9/src/tirex/api_adapter/forecast.py

@@ -0,0 +1,209 @@
+# Copyright (c) NXAI GmbH.
+# This software may be used and distributed according to the terms of the NXAI Community License Agreement.
+
+from abc import ABC, abstractmethod
+from typing import Literal
+
+import torch
+
+from .standard_adapter import ContextType, get_batches
+
+try:
+    from .gluon import format_gluonts_output, get_gluon_batches
+
+    _GLUONTS_AVAILABLE = True
+except ImportError:
+    _GLUONTS_AVAILABLE = False
+
+try:
+    from .hf_data import get_hfdata_batches
+
+    _HF_DATASETS_AVAILABLE = True
+except ImportError:
+    _HF_DATASETS_AVAILABLE = False
+
+
+DEF_TARGET_COLUMN = "target"
+DEF_META_COLUMNS = ("start", "item_id")
+
+
+def _format_output(
+    quantiles: torch.Tensor,
+    means: torch.Tensor,
+    sample_meta: list[dict],
+    quantile_levels: list[float],
+    output_type: Literal["torch", "numpy", "gluonts"],
+):
+    if output_type == "torch":
+        return quantiles.cpu(), means.cpu()
+    elif output_type == "numpy":
+        return quantiles.cpu().numpy(), means.cpu().numpy()
+    elif output_type == "gluonts":
+        if not _GLUONTS_AVAILABLE:
+            raise ValueError("output_type glutonts needs GluonTs but GluonTS is not available (not installed)!")
+        return format_gluonts_output(quantiles, means, sample_meta, quantile_levels)
+    else:
+        raise ValueError(f"Invalid output type: {output_type}")
+
+
+def _as_generator(batches, fc_func, quantile_levels, output_type, **predict_kwargs):
+    for batch_ctx, batch_meta in batches:
+        quantiles, mean = fc_func(batch_ctx, **predict_kwargs)
+        yield _format_output(
+            quantiles=quantiles,
+            means=mean,
+            sample_meta=batch_meta,
+            quantile_levels=quantile_levels,
+            output_type=output_type,
+        )
+
+
+def _gen_forecast(fc_func, batches, output_type, quantile_levels, yield_per_batch, **predict_kwargs):
+    if yield_per_batch:
+        return _as_generator(batches, fc_func, quantile_levels, output_type, **predict_kwargs)
+
+    prediction_q = []
+    prediction_m = []
+    sample_meta = []
+    for batch_ctx, batch_meta in batches:
+        quantiles, mean = fc_func(batch_ctx, **predict_kwargs)
+        prediction_q.append(quantiles)
+        prediction_m.append(mean)
+        sample_meta.extend(batch_meta)
+
+    prediction_q = torch.cat(prediction_q, dim=0)
+    prediction_m = torch.cat(prediction_m, dim=0)
+
+    return _format_output(
+        quantiles=prediction_q,
+        means=prediction_m,
+        sample_meta=sample_meta,
+        quantile_levels=quantile_levels,
+        output_type=output_type,
+    )
+
+
+def _common_forecast_doc():
+    common_doc = f"""
+        This method takes historical context data as input and outputs probabilistic forecasts.
+
+        Args:
+            output_type (Literal["torch", "numpy", "gluonts"], optional):
+                Specifies the desired format of the returned forecasts:
+                - "torch": Returns forecasts as `torch.Tensor` objects [batch_dim, forecast_len, |quantile_levels|]
+                - "numpy": Returns forecasts as `numpy.ndarray` objects [batch_dim, forecast_len, |quantile_levels|]
+                - "gluonts": Returns forecasts as a list of GluonTS `Forecast` objects.
+                Defaults to "torch".
+
+            batch_size (int, optional): The number of time series instances to process concurrently by the model.
+                                        Defaults to 512. Must be $>= 1$.
+
+            quantile_levels (List[float], optional): Quantile levels for which predictions should be generated.
+                                                     Defaults to (0.1, 0.2, ..., 0.9).
+
+            yield_per_batch (bool, optional): If `True`, the method will act as a generator, yielding
+                                              forecasts batch by batch as they are computed.
+                                              Defaults to `False`.
+
+            **predict_kwargs: Additional keyword arguments that are passed directly to the underlying
+                              prediction mechanism of the pre-trained model. Refer to the model's
+                              internal prediction method documentation for available options.
+
+        Returns:
+            The return type depends on `output_type` and `yield_per_batch`:
+                - If `yield_per_batch` is `True`: An iterator that yields forecasts. Each yielded item
+                  will correspond to a batch of forecasts in the format specified by `output_type`.
+                - If `yield_per_batch` is `False`: A single object containing all forecasts.
+                  - If `output_type="torch"`: `Tuple[torch.Tensor, torch.Tensor]` (quantiles, mean).
+                  - If `output_type="numpy"`: `Tuple[numpy.ndarray, numpy.ndarray]` (quantiles, mean).
+                  - If `output_type="gluonts"`: A `List[gluonts.model.forecast.Forecast]` of all forecasts.
+        """
+    return common_doc
+
+
+class ForecastModel(ABC):
+    @abstractmethod
+    def _forecast_quantiles(self, batch, **predict_kwargs):
+        pass
+
+    def forecast(
+        self,
+        context: ContextType,
+        output_type: Literal["torch", "numpy", "gluonts"] = "torch",
+        batch_size: int = 512,
+        quantile_levels: list[float] = (0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9),
+        yield_per_batch: bool = False,
+        **predict_kwargs,
+    ):
+        f"""
+        {_common_forecast_doc}
+        Args:
+            context (ContextType): The historical "context" data of the time series:
+                - `torch.Tensor`: 1D `[context_length]` or 2D `[batch_dim, context_length]` tensor
+                - `np.ndarray`: 1D `[context_length]` or 2D `[batch_dim, context_length]` array
+                - `List[torch.Tensor]`: List of 1D tensors (samples with different lengths get padded per batch)
+                - `List[np.ndarray]`: List of 1D arrays (samples with different lengths get padded per batch)
+        """
+        assert batch_size >= 1, "Batch size must be >= 1"
+        batches = get_batches(context, batch_size)
+        return _gen_forecast(
+            self._forecast_quantiles, batches, output_type, quantile_levels, yield_per_batch, **predict_kwargs
+        )
+
+    def forecast_gluon(
+        self,
+        gluonDataset,
+        output_type: Literal["torch", "numpy", "gluonts"] = "torch",
+        batch_size: int = 512,
+        quantile_levels: list[float] = (0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9),
+        yield_per_batch: bool = False,
+        data_kwargs: dict = {},
+        **predict_kwargs,
+    ):
+        f"""
+        {_common_forecast_doc()}
+
+        Args:
+            gluonDataset (gluon_ts.dataset.common.Dataset): A GluonTS dataset object containing the
+                                                            historical time series data.
+
+            data_kwargs (dict, optional): Additional keyword arguments passed to the
+                                          autogluon data processing function.
+        """
+        assert batch_size >= 1, "Batch size must be >= 1"
+        if not _GLUONTS_AVAILABLE:
+            raise ValueError("forecast_gluon glutonts needs GluonTs but GluonTS is not available (not installed)!")
+        batches = get_gluon_batches(gluonDataset, batch_size, **data_kwargs)
+        return _gen_forecast(
+            self._forecast_quantiles, batches, output_type, quantile_levels, yield_per_batch, **predict_kwargs
+        )
+
+    def forecast_hfdata(
+        self,
+        hf_dataset,
+        output_type: Literal["torch", "numpy", "gluonts"] = "torch",
+        batch_size: int = 512,
+        quantile_levels: list[float] = (0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9),
+        yield_per_batch: bool = False,
+        data_kwargs: dict = {},
+        **predict_kwargs,
+    ):
+        f"""
+        {_common_forecast_doc()}
+
+        Args:
+            hf_dataset (datasets.Dataset): A Hugging Face `Dataset` object containing the
+                                           historical time series data.
+
+            data_kwargs (dict, optional): Additional keyword arguments passed to the
+                                          datasets data processing function.
+        """
+        assert batch_size >= 1, "Batch size must be >= 1"
+        if not _HF_DATASETS_AVAILABLE:
+            raise ValueError(
+                "forecast_hfdata glutonts needs HuggingFace datasets but datasets is not available (not installed)!"
+            )
+        batches = get_hfdata_batches(hf_dataset, batch_size, **data_kwargs)
+        return _gen_forecast(
+            self._forecast_quantiles, batches, output_type, quantile_levels, yield_per_batch, **predict_kwargs
+        )