Views
No views yet


pip install toto-ts1import torch
2from toto.data.util.dataset import MaskedTimeseries
3from toto.inference.forecaster import TotoForecaster
4from toto.model.toto import Toto
5
6DEVICE = 'cuda'
7
8# Load pre-trained Toto model
9toto = Toto.from_pretrained('Datadog/Toto-Open-Base-1.0').to(DEVICE)
10
11# Optional: compile model for enhanced speed
12toto.compile()
13
14forecaster = TotoForecaster(toto.model)
15
16# Example input series (7 variables, 4096 timesteps)
17input_series = torch.randn(7, 4096).to(DEVICE)
18timestamp_seconds = torch.zeros(7, 4096).to(DEVICE)
19time_interval_seconds = torch.full((7,), 60*15).to(DEVICE)
20
21inputs = MaskedTimeseries(
22 series=input_series,
23 padding_mask=torch.full_like(input_series, True, dtype=torch.bool),
24 id_mask=torch.zeros_like(input_series),
25 timestamp_seconds=timestamp_seconds,
26 time_interval_seconds=time_interval_seconds,
27)
28
29# Generate forecasts for next 336 timesteps
30forecast = forecaster.forecast(
31 inputs,
32 prediction_length=336,
33 num_samples=256,
34 samples_per_batch=256,
35)
36
37# Access results
38median_prediction = forecast.median
39prediction_samples = forecast.samples
40lower_quantile = forecast.quantile(0.1)
41upper_quantile = forecast.quantile(0.9)| Checkpoint | Parameters | Config | Size | Notes |
|---|---|---|---|---|
| Toto-Open-Base-1.0 | 151M | Config | 605 MB | Initial release with SOTA performance |
1@inproceedings{
2cohen2026this,
3title={This Time is Different: An Observability Perspective on Time Series Foundation Models},
4author={Ben Cohen and Emaad Khwaja and Youssef Doubli and Salahidine Lemaachi and Chris Lettieri and Charles Masson and Hugo Miccinilli and Elise Ram{\'e} and Qiqi Ren and Afshin Rostamizadeh and Jean Ogier du Terrail and Anna-Monica Toon and Kan Wang and Stephan Xie and Zongzhe Xu and Viktoriya Zhukova and David Asker and Ameet Talwalkar and Othmane Abou-Amal},
5booktitle={The Thirty-ninth Annual Conference on Neural Information Processing Systems},
6year={2026},
7url={https://openreview.net/forum?id=1jDAYXfcS2}
8}