Foundation Models Package (openstef_foundation_models)#

Pretrained foundation-model forecasters that run on an ONNX inference runtime and plug into openstef-models workflows and openstef-beam backtesting. Chronos-2 is the first model family the package ships; the inference backend, provider selection, checkpoint catalog, and workflow machinery are model-agnostic.

The package is split into four subpackages:

  • models holds the checkpoint catalog (Chronos2, CheckpointVariant), the checkpoint resolution types, and the Chronos2Forecaster.

  • presets holds ForecastingWorkflowConfig and create_forecasting_workflow(), the one-call path from config to a runnable workflow.

  • inference holds the InferenceBackend protocol, the ONNX backend, the execution-provider configs, and the host-aware provider selection policy.

  • integrations holds FoundationModelBacktestForecaster, the beam backtesting adapter.

models

Foundation-model artifacts: checkpoints and forecasters.

presets

Presets for building foundation-model forecasters from declarative config.

inference

Inference backends for foundation-model forecasters.

integrations

Integrations with neighbouring OpenSTEF frameworks, one module per framework.