Simons-178M
Browse filesCo-authored-by: RuiLGTM <RuiLGTM@users.noreply.huggingface.co>
Co-authored-by: jiyuantan <jiyuantan@users.noreply.huggingface.co>
- .gitattributes +39 -0
- LICENSE +51 -0
- NOTICE.md +68 -0
- README.md +155 -0
- assets/abel-color.png +0 -0
- assets/abel-white.png +0 -0
- assets/benchmark-places-dark.png +3 -0
- assets/benchmark-places.png +3 -0
- assets/inference-time-dark.png +3 -0
- assets/inference-time.png +3 -0
- config.json +38 -0
- model.safetensors +3 -0
.gitattributes
ADDED
|
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
| 6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
| 7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
| 8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
| 13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
| 14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
| 15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
| 16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
| 18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
| 19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
| 20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
| 21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
| 25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
| 26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
| 27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
| 28 |
+
*.tar filter=lfs diff=lfs merge=lfs -text
|
| 29 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
| 30 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
| 31 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
| 32 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
| 33 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
assets/inference-time.png filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
assets/inference-time-dark.png filter=lfs diff=lfs merge=lfs -text
|
| 38 |
+
assets/benchmark-places.png filter=lfs diff=lfs merge=lfs -text
|
| 39 |
+
assets/benchmark-places-dark.png filter=lfs diff=lfs merge=lfs -text
|
LICENSE
ADDED
|
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Copyright (c) 2026 Abel AI
|
| 2 |
+
|
| 3 |
+
OpenMDW License Agreement, version 1.1 (OpenMDW-1.1)
|
| 4 |
+
|
| 5 |
+
By exercising rights granted to you under this agreement, you accept and agree
|
| 6 |
+
to its terms.
|
| 7 |
+
|
| 8 |
+
As used in this agreement, "Model Materials" means the materials provided to
|
| 9 |
+
you under this agreement, consisting of: (1) one or more machine learning
|
| 10 |
+
models (including architecture and parameters); and (2) all related artifacts
|
| 11 |
+
(including associated data, documentation and software) that are provided to
|
| 12 |
+
you hereunder.
|
| 13 |
+
|
| 14 |
+
Subject to your compliance with this agreement, permission is hereby granted,
|
| 15 |
+
free of charge, to deal in the Model Materials without restriction, including
|
| 16 |
+
under all copyright, patent, database, and trade secret rights included or
|
| 17 |
+
embodied therein.
|
| 18 |
+
|
| 19 |
+
If you distribute any portion of the Model Materials, you shall retain in your
|
| 20 |
+
distribution (1) a copy of this agreement, and (2) all copyright notices and
|
| 21 |
+
other notices of origin included in the Model Materials that are applicable to
|
| 22 |
+
your distribution.
|
| 23 |
+
|
| 24 |
+
If you file, maintain, or voluntarily participate in a lawsuit against any
|
| 25 |
+
person or entity asserting that the Model Materials directly or indirectly
|
| 26 |
+
infringe any patent or copyright, then all rights and grants made to you
|
| 27 |
+
hereunder are terminated, unless that lawsuit was in response to a
|
| 28 |
+
corresponding lawsuit first brought against you.
|
| 29 |
+
|
| 30 |
+
This agreement does not impose any restrictions or obligations with respect to
|
| 31 |
+
any use, modification, or sharing of any outputs generated by using the Model
|
| 32 |
+
Materials.
|
| 33 |
+
|
| 34 |
+
THE MODEL MATERIALS ARE PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
| 35 |
+
OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
| 36 |
+
FITNESS FOR A PARTICULAR PURPOSE, TITLE, NONINFRINGEMENT, ACCURACY, OR THE
|
| 37 |
+
ABSENCE OF LATENT OR OTHER DEFECTS OR ERRORS, WHETHER OR NOT DISCOVERABLE, ALL
|
| 38 |
+
TO THE GREATEST EXTENT PERMISSIBLE UNDER APPLICABLE LAW.
|
| 39 |
+
|
| 40 |
+
YOU ARE SOLELY RESPONSIBLE FOR (1) CLEARING RIGHTS OF OTHER PERSONS THAT MAY
|
| 41 |
+
APPLY TO THE MODEL MATERIALS OR ANY USE THEREOF, INCLUDING WITHOUT LIMITATION
|
| 42 |
+
ANY PERSON'S COPYRIGHTS OR OTHER RIGHTS INCLUDED OR EMBODIED IN THE MODEL
|
| 43 |
+
MATERIALS; (2) OBTAINING ANY NECESSARY CONSENTS, PERMISSIONS OR OTHER RIGHTS
|
| 44 |
+
REQUIRED FOR ANY USE OF THE MODEL MATERIALS; OR (3) PERFORMING ANY DUE
|
| 45 |
+
DILIGENCE OR UNDERTAKING ANY OTHER INVESTIGATIONS INTO THE MODEL MATERIALS OR
|
| 46 |
+
ANYTHING INCORPORATED OR EMBODIED THEREIN.
|
| 47 |
+
|
| 48 |
+
IN NO EVENT SHALL THE PROVIDERS OF THE MODEL MATERIALS BE LIABLE FOR ANY CLAIM,
|
| 49 |
+
DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
|
| 50 |
+
OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE MODEL MATERIALS, THE
|
| 51 |
+
USE THEREOF OR OTHER DEALINGS THEREIN.
|
NOTICE.md
ADDED
|
@@ -0,0 +1,68 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Notices
|
| 2 |
+
|
| 3 |
+
Simons, Copyright 2026 Abel AI Lab (https://abel.ai).
|
| 4 |
+
|
| 5 |
+
## Notices required by the training data's licenses
|
| 6 |
+
|
| 7 |
+
Each training dataset of the Simons models keeps its own license and terms; the data were cleaned,
|
| 8 |
+
resampled and cut into training windows.
|
| 9 |
+
|
| 10 |
+
- ERA5: Contains modified Copernicus Climate Change Service information 2026. Neither the European
|
| 11 |
+
Commission nor ECMWF is responsible for any use that may be made of the Copernicus information
|
| 12 |
+
or data it contains.
|
| 13 |
+
- CAMS European air-quality forecasts: Contains modified Copernicus Atmosphere Monitoring Service
|
| 14 |
+
information 2026. Neither the European Commission nor ECMWF is responsible for any use that may
|
| 15 |
+
be made of the Copernicus information or data it contains.
|
| 16 |
+
- ECMWF IFS weather forecasts, retrieved through the Open-Meteo archive: Weather data by
|
| 17 |
+
Open-Meteo.com (https://open-meteo.com/), CC BY 4.0. Based on data and products of the European
|
| 18 |
+
Centre for Medium-Range Weather Forecasts (ECMWF, https://www.ecmwf.int/), CC BY 4.0.
|
| 19 |
+
- NASA POWER: Weather data were obtained from the NASA Langley Research Center POWER Project
|
| 20 |
+
funded through the NASA Earth Science Directorate Applied Science Program.
|
| 21 |
+
- NCEP/CPC L3 Half Hourly 4km Global Merged IR (GPM_MERGIR): NASA is the source of these data
|
| 22 |
+
(https://disc.gsfc.nasa.gov/datasets/GPM_MERGIR_1/summary).
|
| 23 |
+
- Elexon BMRS: Contains BMRS data © Elexon Limited copyright and database right 2026.
|
| 24 |
+
- National Highways WebTRIS: Contains public sector information licensed under the Open Government
|
| 25 |
+
Licence v3.0.
|
| 26 |
+
- NESO Data Portal: Supported by National Energy SO Open Data.
|
| 27 |
+
- EirGrid: Supported by EirGrid Group Data.
|
| 28 |
+
- Retrosheet: The information used here was obtained free of charge from and is copyrighted by
|
| 29 |
+
Retrosheet. Interested parties may contact Retrosheet at "www.retrosheet.org".
|
| 30 |
+
- WorldTrace: Contains information from
|
| 31 |
+
[WorldTrace (UniTraj)](https://huggingface.co/datasets/OpenTrace/WorldTrace), which is made
|
| 32 |
+
available under the
|
| 33 |
+
[Open Database License (ODbL) 1.0](https://opendatacommons.org/licenses/odbl/1-0/). Map
|
| 34 |
+
matching uses data © [OpenStreetMap contributors](https://www.openstreetmap.org/copyright). The
|
| 35 |
+
WorldTrace-derived training data are offered under the ODbL 1.0 in
|
| 36 |
+
[Simons-Pretrain](https://huggingface.co/datasets/abel-lab/Simons-Pretrain)
|
| 37 |
+
(`univariate/worldtrace/`).
|
| 38 |
+
- Banco Central do Brasil Focus survey: Contains information from the
|
| 39 |
+
[Expectativas de Mercado dataset](https://dadosabertos.bcb.gov.br/dataset/expectativas-mercado)
|
| 40 |
+
(Focus survey) of Banco Central do Brasil, which is made available under the
|
| 41 |
+
[Open Database License (ODbL) 1.0](https://opendatacommons.org/licenses/odbl/1-0/). Fonte: Banco
|
| 42 |
+
Central do Brasil, Sistema de Expectativas de Mercado. The derived data are offered under the ODbL
|
| 43 |
+
1.0 in [Simons-Pretrain](https://huggingface.co/datasets/abel-lab/Simons-Pretrain)
|
| 44 |
+
(`covariates/bcb_focus_inflation/`).
|
| 45 |
+
- opentransportdata.swiss: Source of the raw data: opentransportdata.swiss.
|
| 46 |
+
- RTE eCO2mix: Data from RTE éCO2mix (https://odre.opendatasoft.com/), Licence Ouverte 2.0.
|
| 47 |
+
- CausalRivers: Data from CausalRivers (https://causalrivers.github.io/), licensed under
|
| 48 |
+
dl-de/by-2-0 (https://www.govdata.de/dl-de/by-2-0).
|
| 49 |
+
- AEMO: Contains data from the Australian Energy Market Operator (AEMO), used under AEMO's
|
| 50 |
+
copyright permissions.
|
| 51 |
+
- Elia: Contains data from Elia Open Data (https://opendata.elia.be/).
|
| 52 |
+
- LargeAQ: from CauAir (https://github.com/PoorOtterBob/CauAir), CC BY 4.0. Ma et al., "Causal
|
| 53 |
+
Learning Meet Covariates: Empowering Lightweight and Effective Nationwide Air Quality
|
| 54 |
+
Forecasting", IJCAI 2025 (https://www.ijcai.org/proceedings/2025/353).
|
| 55 |
+
- CC BY-SA 4.0: training data from
|
| 56 |
+
[Building Data Genome Project 2](https://github.com/buds-lab/building-data-genome-project-2),
|
| 57 |
+
[Open D1NAMO](https://zenodo.org/records/5651217),
|
| 58 |
+
[NMT Scalp EEG Dataset](https://dll.seecs.nust.edu.pk/downloads/)
|
| 59 |
+
([Khan et al. 2022](https://doi.org/10.3389/fnins.2021.755817)), the
|
| 60 |
+
[HAI industrial control system dataset](https://github.com/icsdataset/hai) and
|
| 61 |
+
[Wikipedia](https://en.wikipedia.org/) club-season, episode-list and team-season articles
|
| 62 |
+
(Wikipedia contributors; the team-season articles give the NBA, NHL and WNBA event dates in
|
| 63 |
+
`nyc_subway_events`), licensed under
|
| 64 |
+
[CC BY-SA 4.0](https://creativecommons.org/licenses/by-sa/4.0/); the redistributed copies are in
|
| 65 |
+
[Simons-Pretrain](https://huggingface.co/datasets/abel-lab/Simons-Pretrain).
|
| 66 |
+
- The datasets in Simons-Pretrain are credited, with their licenses and links to their
|
| 67 |
+
sources, in its
|
| 68 |
+
[NOTICE.md](https://huggingface.co/datasets/abel-lab/Simons-Pretrain/blob/main/NOTICE.md).
|
README.md
ADDED
|
@@ -0,0 +1,155 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: openmdw-1.1
|
| 3 |
+
library_name: simons
|
| 4 |
+
pipeline_tag: time-series-forecasting
|
| 5 |
+
datasets:
|
| 6 |
+
- abel-lab/Simons-Pretrain
|
| 7 |
+
thumbnail: https://huggingface.co/abel-lab/Simons-178M/resolve/main/assets/abel-color.png
|
| 8 |
+
tags:
|
| 9 |
+
- time-series
|
| 10 |
+
- forecasting
|
| 11 |
+
- probabilistic-forecasting
|
| 12 |
+
- foundation-model
|
| 13 |
+
- covariates
|
| 14 |
+
- pretrained
|
| 15 |
+
- pytorch
|
| 16 |
+
model-index:
|
| 17 |
+
- name: Simons-178M
|
| 18 |
+
results:
|
| 19 |
+
- task:
|
| 20 |
+
type: time-series-forecasting
|
| 21 |
+
dataset:
|
| 22 |
+
name: GIFT-Eval
|
| 23 |
+
type: Salesforce/GiftEval
|
| 24 |
+
metrics:
|
| 25 |
+
- name: MASE (relative to Seasonal Naive)
|
| 26 |
+
type: mase
|
| 27 |
+
value: 0.6754
|
| 28 |
+
- name: CRPS (relative to Seasonal Naive)
|
| 29 |
+
type: crps
|
| 30 |
+
value: 0.4609
|
| 31 |
+
source:
|
| 32 |
+
name: GIFT-Eval (scored with the benchmark code)
|
| 33 |
+
url: https://github.com/SalesforceAIResearch/gift-eval
|
| 34 |
+
---
|
| 35 |
+
|
| 36 |
+
<p align="center">
|
| 37 |
+
<img class="dark:hidden"
|
| 38 |
+
src="https://huggingface.co/abel-lab/Simons-178M/resolve/main/assets/abel-color.png" alt="Abel"
|
| 39 |
+
width="240" />
|
| 40 |
+
<img class="hidden dark:block"
|
| 41 |
+
src="https://huggingface.co/abel-lab/Simons-178M/resolve/main/assets/abel-white.png" alt="Abel"
|
| 42 |
+
width="240" />
|
| 43 |
+
</p>
|
| 44 |
+
|
| 45 |
+
# Simons-178M
|
| 46 |
+
|
| 47 |
+
Simons-178M is an open time series foundation model from Abel AI Lab for probabilistic forecasting.
|
| 48 |
+
It supports univariate and multivariate series with known-future covariates.
|
| 49 |
+
The [GitHub repository](https://github.com/Abel-ai-lab/Simons) contains inference and training code.
|
| 50 |
+
All four sizes and the training data are in the [Simons 1](https://huggingface.co/collections/abel-lab/simons-1-6ae3d4ce9705938e293c52dd) collection.
|
| 51 |
+
The [Simons Forecast Explorer](https://huggingface.co/spaces/abel-lab/simons-forecast-explorer) shows forecasts of the four sizes in the browser.
|
| 52 |
+
|
| 53 |
+
## Model details
|
| 54 |
+
|
| 55 |
+
- **Size:** 177.7M parameters.
|
| 56 |
+
- **Architecture:** a causal transformer over patches of 32 time steps, then an attention layer
|
| 57 |
+
across the variates of a series; 12 layers, width 1024, 16 heads.
|
| 58 |
+
- **Context and horizon:** up to 32,768 time steps of history; 512 steps per forward pass, longer
|
| 59 |
+
horizons block by block.
|
| 60 |
+
- **Inputs:** up to 32 variates and up to 16 known-future covariates per series, numeric or
|
| 61 |
+
categorical; NaN marks missing values. Past-only covariates are accepted and dropped.
|
| 62 |
+
- **Outputs:** 9 quantiles (0.1, 0.2, ..., 0.9).
|
| 63 |
+
- **Training:** four stages, from synthetic pretraining to post-training on real data; the released
|
| 64 |
+
weights are the uniform average of the two continuations (Model soups).
|
| 65 |
+
- **Inference:** PyTorch on CPU or GPU, with the `simons` package, version 0.1.0 or later.
|
| 66 |
+
|
| 67 |
+
## Benchmarks
|
| 68 |
+
|
| 69 |
+
<p align="center">
|
| 70 |
+
<img class="dark:hidden"
|
| 71 |
+
src="https://huggingface.co/abel-lab/Simons-178M/resolve/main/assets/benchmark-places.png" alt="Places on public benchmark leaderboards"
|
| 72 |
+
width="720" />
|
| 73 |
+
<img class="hidden dark:block"
|
| 74 |
+
src="https://huggingface.co/abel-lab/Simons-178M/resolve/main/assets/benchmark-places-dark.png" alt="Places on public benchmark leaderboards"
|
| 75 |
+
width="720" />
|
| 76 |
+
</p>
|
| 77 |
+
|
| 78 |
+
The place of Simons-297M and four widely used models on each benchmark's own leaderboard. On
|
| 79 |
+
GIFT-Eval, fev-bench and TIME the places are among the open-weight entries, from the snapshots used
|
| 80 |
+
for this release (GIFT-Eval: October 30, 2026).
|
| 81 |
+
|
| 82 |
+
### Efficiency
|
| 83 |
+
|
| 84 |
+
Simons is fast to run: on one A100 GPU, Simons-297M forecasts a whole benchmark in minutes, about ten
|
| 85 |
+
times faster than models of similar size.
|
| 86 |
+
|
| 87 |
+
<p align="center">
|
| 88 |
+
<img class="dark:hidden"
|
| 89 |
+
src="https://huggingface.co/abel-lab/Simons-178M/resolve/main/assets/inference-time.png" alt="Time to forecast GIFT-Eval, TIME and fev-bench"
|
| 90 |
+
width="720" />
|
| 91 |
+
<img class="hidden dark:block"
|
| 92 |
+
src="https://huggingface.co/abel-lab/Simons-178M/resolve/main/assets/inference-time-dark.png" alt="Time to forecast GIFT-Eval, TIME and fev-bench"
|
| 93 |
+
width="720" />
|
| 94 |
+
</p>
|
| 95 |
+
|
| 96 |
+
## Get started
|
| 97 |
+
|
| 98 |
+
```shell
|
| 99 |
+
pip install "git+https://github.com/Abel-ai-lab/Simons"
|
| 100 |
+
```
|
| 101 |
+
|
| 102 |
+
```python
|
| 103 |
+
import torch
|
| 104 |
+
from simons import Simons
|
| 105 |
+
|
| 106 |
+
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 107 |
+
model = Simons.from_pretrained("abel-lab/Simons-178M", device=device)
|
| 108 |
+
|
| 109 |
+
# [batch, time] or [batch, variates, time]; NaN marks missing values
|
| 110 |
+
history = torch.randn(8, 2000).cumsum(-1)
|
| 111 |
+
quantiles = model.forecast(history, horizon=96) # [8, 96, 9]
|
| 112 |
+
|
| 113 |
+
# Known-future covariates: [batch, covariates, time + horizon]
|
| 114 |
+
covariates = torch.randn(8, 2, 2000 + 96)
|
| 115 |
+
quantiles = model.forecast(history, horizon=96, covariates=covariates)
|
| 116 |
+
```
|
| 117 |
+
|
| 118 |
+
- **Training data:** [abel-lab/Simons-Pretrain](https://huggingface.co/datasets/abel-lab/Simons-Pretrain),
|
| 119 |
+
the public part of the data the models were trained on.
|
| 120 |
+
- **Getting started:** [docs/GETTING_STARTED.md](https://github.com/Abel-ai-lab/Simons/blob/main/docs/GETTING_STARTED.md) covers the input format,
|
| 121 |
+
covariates, forecast options and fine-tuning.
|
| 122 |
+
- **Models:** [docs/MODEL.md](https://github.com/Abel-ai-lab/Simons/blob/main/docs/MODEL.md) compares the four sizes and gives their scores.
|
| 123 |
+
- **Training:** [docs/TRAINING.md](https://github.com/Abel-ai-lab/Simons/blob/main/docs/TRAINING.md) covers the stages, configs and GPU settings.
|
| 124 |
+
- **Data:** [docs/DATA.md](https://github.com/Abel-ai-lab/Simons/blob/main/docs/DATA.md) describes what the models were trained on.
|
| 125 |
+
|
| 126 |
+
## Training data
|
| 127 |
+
|
| 128 |
+
- Synthetic series from the open generators in
|
| 129 |
+
[`src/simons/synth`](https://github.com/Abel-ai-lab/Simons/tree/main/src/simons/synth), with
|
| 130 |
+
synthetic known-future covariates of known effect.
|
| 131 |
+
- Public datasets with a permissive or share-alike license, most of them released as
|
| 132 |
+
[abel-lab/Simons-Pretrain](https://huggingface.co/datasets/abel-lab/Simons-Pretrain).
|
| 133 |
+
- Internal data: an in-house corpus.
|
| 134 |
+
|
| 135 |
+
See the
|
| 136 |
+
[dataset](https://huggingface.co/datasets/abel-lab/Simons-Pretrain) for data details
|
| 137 |
+
and the [training quickstart](https://github.com/Abel-ai-lab/Simons#train) for training commands.
|
| 138 |
+
Notices required by the training data's licenses: [NOTICE.md](NOTICE.md).
|
| 139 |
+
|
| 140 |
+
## License
|
| 141 |
+
|
| 142 |
+
The weights are licensed under OpenMDW-1.1 ([LICENSE](LICENSE)); the code on GitHub is licensed
|
| 143 |
+
under Apache-2.0 ([LICENSE](https://github.com/Abel-ai-lab/Simons/blob/main/LICENSE)). Each training
|
| 144 |
+
dataset keeps its own license; see [NOTICE.md](NOTICE.md) for details.
|
| 145 |
+
|
| 146 |
+
## Citation
|
| 147 |
+
|
| 148 |
+
```bibtex
|
| 149 |
+
@misc{simons2026,
|
| 150 |
+
title = {Simons: Building Open Time Series Foundation Models through Auto-Research},
|
| 151 |
+
author = {{Abel AI Lab}},
|
| 152 |
+
year = {2026},
|
| 153 |
+
url = {https://github.com/Abel-ai-lab/Simons}
|
| 154 |
+
}
|
| 155 |
+
```
|
assets/abel-color.png
ADDED
|
assets/abel-white.png
ADDED
|
assets/benchmark-places-dark.png
ADDED
|
Git LFS Details
|
assets/benchmark-places.png
ADDED
|
Git LFS Details
|
assets/inference-time-dark.png
ADDED
|
Git LFS Details
|
assets/inference-time.png
ADDED
|
Git LFS Details
|
config.json
ADDED
|
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_type": "simons",
|
| 3 |
+
"context_length": 32768,
|
| 4 |
+
"prediction_length": 512,
|
| 5 |
+
"patch_length": 32,
|
| 6 |
+
"width": 1024,
|
| 7 |
+
"layers": 12,
|
| 8 |
+
"heads": 16,
|
| 9 |
+
"near_context": 4096,
|
| 10 |
+
"far_pool": 8,
|
| 11 |
+
"far_min_history": 7680,
|
| 12 |
+
"stats_window": 8192,
|
| 13 |
+
"period_slots": 4,
|
| 14 |
+
"period_alpha": 0.05,
|
| 15 |
+
"period_local": 5.0,
|
| 16 |
+
"detector_mid": 7680,
|
| 17 |
+
"views": true,
|
| 18 |
+
"tail_paths": [
|
| 19 |
+
1,
|
| 20 |
+
4,
|
| 21 |
+
7
|
| 22 |
+
],
|
| 23 |
+
"sign_clip": true,
|
| 24 |
+
"integer_snap": true,
|
| 25 |
+
"covariates": true,
|
| 26 |
+
"binary_rows": true,
|
| 27 |
+
"fit_row": true,
|
| 28 |
+
"past_rows": false,
|
| 29 |
+
"read_after": 5,
|
| 30 |
+
"read_binary": true,
|
| 31 |
+
"xreg": true,
|
| 32 |
+
"xreg_gate": 0.05,
|
| 33 |
+
"xreg_square": false,
|
| 34 |
+
"xreg_guard": true,
|
| 35 |
+
"xreg_guard_margin": 0.9,
|
| 36 |
+
"xreg_guard_min": 0,
|
| 37 |
+
"rows_only_history": 4096
|
| 38 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8629ebf2ecbda13c1d852cd00a4db4c59bab41f96c725916aea0b2a51f52d327
|
| 3 |
+
size 710996944
|