Skip to main content

openstef-foundation-models

Foundation model support for OpenSTEF — bringing pre-trained, ONNX-based forecasting models to the OpenSTEF ecosystem.

Installation

Pick exactly one ONNX runtime — [cpu] and [gpu] are mutually exclusive.

CPU (default — the meta-package openstef installs this for you):

pip install "openstef-foundation-models[cpu]"

GPU (CUDA):

pip install "openstef-foundation-models[gpu]"

Note: Do not install both [cpu] and [gpu] in the same environment — onnxruntime and onnxruntime-gpu collide. They're declared as conflicting extras so uv enforces the choice; pip does not, so pick one yourself.

Selecting a checkpoint

OpenSTEF publishes its checkpoints to the HuggingFace Hub. Pick a model size and variant from the catalog instead of hand-writing repo ids and filenames:

from openstef_foundation_models.models import Chronos2, CheckpointVariant
from openstef_foundation_models.presets import ForecastingWorkflowConfig

# Default: the base Chronos-2, dynamic shapes — runs on any provider.
config = ForecastingWorkflowConfig()

# The smaller model, static shapes — enables the CoreML provider on macOS.
config = ForecastingWorkflowConfig(
    checkpoint=Chronos2.SMALL.checkpoint(CheckpointVariant.STATIC),
)

# Let the host decide: static on macOS, dynamic elsewhere.
config = ForecastingWorkflowConfig(
    checkpoint=Chronos2.BASE.checkpoint(CheckpointVariant.recommended()),
)

Available sizes are Chronos2.BASE (chronos-2) and Chronos2.SMALL (chronos-2-small); variants are DYNAMIC (portable) and STATIC (frozen shapes, the macOS/CoreML path in the default provider fallback chain).

Metadata

Release files for openstef-foundation-models 4.4.6

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for openstef-foundation-models 4.4.6
File Size Uploaded
openstef_foundation_models-4.4.6.tar.gz 37.1 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for openstef-foundation-models 4.4.6
File Interpreter ABI Platform
openstef_foundation_models-4.4.6-py3-none-any.whl Python 3 none any Details

Total release size: 66.3 kB

Release files / openstef_foundation_models-4.4.6.tar.gz

Download URL openstef_foundation_models-4.4.6.tar.gz
Size 37.1 kB
Tags Source
SHA-256 checksum
How to use checksums
3d170c1147a4535b6def2c2eaa4f444a34135337f8397f3f8a79d7f44c923e93
BLAKE2b-256 checksum
How to use checksums
ed02c1b5bc474b1cdddf7ad170657c0e33d648eb0a23b03839ec4f2d432e3028
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.12.21 {"installer":{"name":"uv","version":"0.12.21","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Sep 30, 2026.

Transparency log

Release files / openstef_foundation_models-4.4.6-py3-none-any.whl

Download URL openstef_foundation_models-4.4.6-py3-none-any.whl
Size 29.2 kB
Tags Python 3
SHA-256 checksum
How to use checksums
759febbf269609e19e050ef1bc920cafa3b35fcccec59b38bda45d864f76be1b
BLAKE2b-256 checksum
How to use checksums
0fdcf9f6c918fc0adb33fb057de334294e29a523d2085852ca34cd4e12810d26
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.12.21 {"installer":{"name":"uv","version":"0.12.21","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Sep 30, 2026.

Transparency log

Release history Release notifications | RSS feed

4.4.8

2 release files

4.4.7

2 release files

This release

4.4.6 This release

2 release files

4.4.5

2 release files

4.4.4

2 release files

4.4.3

2 release files

4.4.2

2 release files

4.4.1

2 release files

4.3.1

2 release files

4.3.0

2 release files

4.2.3

2 release files

4.2.2

2 release files

4.2.1

2 release files

4.1.2

2 release files

4.1.1

2 release files

4.1.0

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page