chronos-forecasting 2.2.2__tar.gz → 2.3.1__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.github/ISSUE_TEMPLATE/bug_report.md +1 -1
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/PKG-INFO +20 -16
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/README.md +9 -3
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/notebooks/chronos-2-quickstart.ipynb +105 -58
- chronos_forecasting-2.3.1/notebooks/deploy-chronos-to-amazon-sagemaker.ipynb +1435 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/pyproject.toml +10 -12
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/train.py +3 -1
- chronos_forecasting-2.3.1/src/chronos/__about__.py +1 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/base.py +122 -98
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/boto_utils.py +12 -5
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos2/config.py +4 -1
- chronos_forecasting-2.3.1/src/chronos/chronos2/dataset.py +373 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos2/layers.py +62 -17
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos2/model.py +37 -18
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos2/pipeline.py +152 -177
- chronos_forecasting-2.3.1/src/chronos/chronos2/preprocess.py +686 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos_bolt.py +48 -11
- chronos_forecasting-2.3.1/src/chronos/df_utils.py +338 -0
- chronos_forecasting-2.3.1/test/dummy-chronos-model/model.safetensors +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/test_chronos2.py +226 -58
- chronos_forecasting-2.3.1/test/test_df_utils.py +369 -0
- chronos_forecasting-2.3.1/test/test_preprocess.py +575 -0
- chronos_forecasting-2.2.2/notebooks/deploy-chronos-to-amazon-sagemaker.ipynb +0 -1173
- chronos_forecasting-2.2.2/src/chronos/__about__.py +0 -1
- chronos_forecasting-2.2.2/src/chronos/chronos2/dataset.py +0 -655
- chronos_forecasting-2.2.2/src/chronos/df_utils.py +0 -323
- chronos_forecasting-2.2.2/test/dummy-chronos-model/pytorch_model.bin +0 -0
- chronos_forecasting-2.2.2/test/test_df_utils.py +0 -361
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.gitattributes +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.github/ISSUE_TEMPLATE/config.yml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.github/workflows/ci.yml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.github/workflows/eval-model.yml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.github/workflows/publish-to-pypi.yml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.gitignore +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/CITATION.cff +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/CODE_OF_CONDUCT.md +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/CONTRIBUTING.md +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/LICENSE +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/NOTICE +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/ci/evaluate/backtest_config.yaml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/README.md +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/agg-relative-score.py +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/configs/in-domain.yaml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/configs/zero-shot.yaml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/evaluate.py +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-base-agg-rel-scores.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-base-in-domain.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-base-zero-shot.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-mini-agg-rel-scores.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-mini-in-domain.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-mini-zero-shot.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-small-agg-rel-scores.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-small-in-domain.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-small-zero-shot.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-tiny-agg-rel-scores.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-tiny-in-domain.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-tiny-zero-shot.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-base-agg-rel-scores.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-base-in-domain.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-base-zero-shot.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-large-agg-rel-scores.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-large-in-domain.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-large-zero-shot.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-mini-agg-rel-scores.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-mini-in-domain.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-mini-zero-shot.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-small-agg-rel-scores.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-small-in-domain.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-small-zero-shot.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-tiny-agg-rel-scores.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-tiny-in-domain.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-tiny-zero-shot.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/seasonal-naive-in-domain.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/seasonal-naive-zero-shot.csv +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/kernel-synth.py +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/configs/chronos-gpt2.yaml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/configs/chronos-t5-base.yaml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/configs/chronos-t5-large.yaml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/configs/chronos-t5-mini.yaml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/configs/chronos-t5-small.yaml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/configs/chronos-t5-tiny.yaml +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/__init__.py +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos.py +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos2/__init__.py +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos2/trainer.py +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/utils.py +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/__init__.py +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos-bolt-model/config.json +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos-bolt-model/model.safetensors +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos-model/config.json +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos-model/generation_config.json +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos2-lora/adapter_config.json +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos2-lora/adapter_model.safetensors +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos2-model/config.json +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos2-model/model.safetensors +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/test_chronos.py +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/test_chronos_bolt.py +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/test_utils.py +0 -0
- {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/util.py +0 -0
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: chronos-forecasting
|
|
3
|
-
Version: 2.
|
|
3
|
+
Version: 2.3.1
|
|
4
4
|
Summary: Chronos: Pretrained models for time series forecasting
|
|
5
5
|
Project-URL: Homepage, https://github.com/amazon-science/chronos-forecasting
|
|
6
6
|
Project-URL: Issues, https://github.com/amazon-science/chronos-forecasting/issues
|
|
@@ -189,32 +189,30 @@ Classifier: Operating System :: OS Independent
|
|
|
189
189
|
Classifier: Programming Language :: Python :: 3
|
|
190
190
|
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
|
|
191
191
|
Requires-Python: >=3.10
|
|
192
|
-
Requires-Dist: accelerate<2,>=0
|
|
192
|
+
Requires-Dist: accelerate<2,>=1.1.0
|
|
193
193
|
Requires-Dist: einops<1,>=0.7.0
|
|
194
194
|
Requires-Dist: numpy<3,>=1.21
|
|
195
|
-
Requires-Dist:
|
|
195
|
+
Requires-Dist: pandas<4,>=2.0
|
|
196
196
|
Requires-Dist: torch<3,>=2.2
|
|
197
|
-
Requires-Dist: transformers<
|
|
197
|
+
Requires-Dist: transformers<6,>=4.41
|
|
198
198
|
Provides-Extra: dev
|
|
199
|
-
Requires-Dist: datasets<
|
|
200
|
-
Requires-Dist: fev
|
|
199
|
+
Requires-Dist: datasets<5,>=3.0; extra == 'dev'
|
|
200
|
+
Requires-Dist: fev<1,>=0.8.0; extra == 'dev'
|
|
201
201
|
Requires-Dist: gluonts[pro]~=0.16; extra == 'dev'
|
|
202
202
|
Requires-Dist: joblib; extra == 'dev'
|
|
203
|
-
Requires-Dist: pandas<2.4,>=2.0; extra == 'dev'
|
|
204
203
|
Requires-Dist: rich; extra == 'dev'
|
|
205
204
|
Requires-Dist: tensorboard; extra == 'dev'
|
|
206
205
|
Requires-Dist: typer; extra == 'dev'
|
|
207
206
|
Requires-Dist: typer-config; extra == 'dev'
|
|
208
207
|
Provides-Extra: extras
|
|
209
208
|
Requires-Dist: boto3<2,>=1.10; extra == 'extras'
|
|
210
|
-
Requires-Dist: fev
|
|
211
|
-
Requires-Dist: pandas[pyarrow]<
|
|
212
|
-
Requires-Dist: peft<0.18
|
|
209
|
+
Requires-Dist: fev<1,>=0.8.0; extra == 'extras'
|
|
210
|
+
Requires-Dist: pandas[pyarrow]<4,>=2.0; extra == 'extras'
|
|
211
|
+
Requires-Dist: peft<1,>=0.18.1; extra == 'extras'
|
|
213
212
|
Provides-Extra: test
|
|
214
213
|
Requires-Dist: boto3<2,>=1.10; extra == 'test'
|
|
215
|
-
Requires-Dist: fev
|
|
216
|
-
Requires-Dist:
|
|
217
|
-
Requires-Dist: peft<1,>=0.13.0; extra == 'test'
|
|
214
|
+
Requires-Dist: fev<1,>=0.8.0; extra == 'test'
|
|
215
|
+
Requires-Dist: peft<1,>=0.18.1; extra == 'test'
|
|
218
216
|
Requires-Dist: pytest~=8.0; extra == 'test'
|
|
219
217
|
Provides-Extra: typecheck
|
|
220
218
|
Requires-Dist: mypy~=1.9; extra == 'typecheck'
|
|
@@ -237,8 +235,9 @@ Description-Content-Type: text/markdown
|
|
|
237
235
|
|
|
238
236
|
|
|
239
237
|
## 🚀 News
|
|
238
|
+
- **5 Jun 2026**: ☁️ Deploy Chronos-2 to AWS with [AutoGluon-Cloud](https://github.com/autogluon/autogluon-cloud) — real-time, serverless, or batch inference in 3 lines of code, pandas DataFrames in, forecasts out. See the [new deployment guide](https://auto.gluon.ai/cloud/stable/tutorials/foundation-model-timeseries.html).
|
|
239
|
+
- **30 Dec 2025**: 🧠 Deploy Chronos-2 to AWS with Amazon SageMaker JumpStart — production-ready real-time endpoints on CPU or GPU, ready to integrate into your existing AWS workflows. See the [deployment tutorial](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb).
|
|
240
240
|
- **20 Oct 2025**: 🚀 [Chronos-2](https://huggingface.co/amazon/chronos-2) released. It offers _zero-shot_ support for univariate, multivariate, and covariate-informed forecasting tasks. Chronos-2 achieves the best performance on fev-bench, GIFT-Eval and Chronos Benchmark II amongst pretrained models. Check out [this notebook](notebooks/chronos-2-quickstart.ipynb) to get started with Chronos-2.
|
|
241
|
-
- **14 Feb 2025**: 🚀 Chronos-Bolt is now available on Amazon SageMaker JumpStart! Check out the [tutorial notebook](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb) to learn how to deploy Chronos endpoints for production use in 3 lines of code.
|
|
242
241
|
- **12 Dec 2024**: 📊 We released [`fev`](https://github.com/autogluon/fev), a lightweight package for benchmarking time series forecasting models based on the [Hugging Face `datasets`](https://huggingface.co/docs/datasets/en/index) library.
|
|
243
242
|
- **26 Nov 2024**: ⚡️ Chronos-Bolt models released [on HuggingFace](https://huggingface.co/collections/amazon/chronos-models-65f1791d630a8d57cb718444). Chronos-Bolt models are more accurate (5% lower error), up to 250x faster and 20x more memory efficient than the original Chronos models of the same size!
|
|
244
243
|
- **13 Mar 2024**: 🚀 Chronos [paper](https://arxiv.org/abs/2403.07815) and inference code released.
|
|
@@ -258,6 +257,8 @@ This package provides an interface to the Chronos family of **pretrained time se
|
|
|
258
257
|
| Model ID | Parameters |
|
|
259
258
|
| ---------------------------------------------------------------------- | ---------- |
|
|
260
259
|
| [`amazon/chronos-2`](https://huggingface.co/amazon/chronos-2) | 120M |
|
|
260
|
+
| [`autogluon/chronos-2-synth`](https://huggingface.co/autogluon/chronos-2-synth) | 120M |
|
|
261
|
+
| [`autogluon/chronos-2-small`](https://huggingface.co/autogluon/chronos-2-small) | 28M |
|
|
261
262
|
| [`amazon/chronos-bolt-tiny`](https://huggingface.co/amazon/chronos-bolt-tiny) | 9M |
|
|
262
263
|
| [`amazon/chronos-bolt-mini`](https://huggingface.co/amazon/chronos-bolt-mini) | 21M |
|
|
263
264
|
| [`amazon/chronos-bolt-small`](https://huggingface.co/amazon/chronos-bolt-small) | 48M |
|
|
@@ -279,7 +280,9 @@ pip install chronos-forecasting
|
|
|
279
280
|
```
|
|
280
281
|
|
|
281
282
|
> [!TIP]
|
|
282
|
-
> For
|
|
283
|
+
> For production use, we recommend deploying Chronos-2 to Amazon SageMaker. Two options:
|
|
284
|
+
> - **[AutoGluon-Cloud](https://auto.gluon.ai/cloud/stable/tutorials/foundation-model-timeseries.html)** (recommended) — high-level Python API. Pandas DataFrames in, forecasts out. Real-time, serverless, and batch inference out of the box.
|
|
285
|
+
> - **[SageMaker JumpStart](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb)** — production-ready real-time endpoints on CPU or GPU, ready to integrate into your existing AWS workflows.
|
|
283
286
|
|
|
284
287
|
|
|
285
288
|
### Forecasting
|
|
@@ -345,7 +348,8 @@ plt.legend()
|
|
|
345
348
|
<a href="https://colab.research.google.com/github/amazon-science/chronos-forecasting/blob/main/notebooks/chronos-2-quickstart.ipynb">
|
|
346
349
|
<img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab" height="18" align="absmiddle">
|
|
347
350
|
</a>
|
|
348
|
-
- [Deploy Chronos-2
|
|
351
|
+
- [Deploy Chronos-2 to AWS with AutoGluon-Cloud](https://auto.gluon.ai/cloud/stable/tutorials/foundation-model-timeseries.html) (recommended)
|
|
352
|
+
- [Deploy Chronos-2 to AWS with SageMaker JumpStart](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb)
|
|
349
353
|
|
|
350
354
|
## 📝 Citation
|
|
351
355
|
|
|
@@ -15,8 +15,9 @@
|
|
|
15
15
|
|
|
16
16
|
|
|
17
17
|
## 🚀 News
|
|
18
|
+
- **5 Jun 2026**: ☁️ Deploy Chronos-2 to AWS with [AutoGluon-Cloud](https://github.com/autogluon/autogluon-cloud) — real-time, serverless, or batch inference in 3 lines of code, pandas DataFrames in, forecasts out. See the [new deployment guide](https://auto.gluon.ai/cloud/stable/tutorials/foundation-model-timeseries.html).
|
|
19
|
+
- **30 Dec 2025**: 🧠 Deploy Chronos-2 to AWS with Amazon SageMaker JumpStart — production-ready real-time endpoints on CPU or GPU, ready to integrate into your existing AWS workflows. See the [deployment tutorial](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb).
|
|
18
20
|
- **20 Oct 2025**: 🚀 [Chronos-2](https://huggingface.co/amazon/chronos-2) released. It offers _zero-shot_ support for univariate, multivariate, and covariate-informed forecasting tasks. Chronos-2 achieves the best performance on fev-bench, GIFT-Eval and Chronos Benchmark II amongst pretrained models. Check out [this notebook](notebooks/chronos-2-quickstart.ipynb) to get started with Chronos-2.
|
|
19
|
-
- **14 Feb 2025**: 🚀 Chronos-Bolt is now available on Amazon SageMaker JumpStart! Check out the [tutorial notebook](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb) to learn how to deploy Chronos endpoints for production use in 3 lines of code.
|
|
20
21
|
- **12 Dec 2024**: 📊 We released [`fev`](https://github.com/autogluon/fev), a lightweight package for benchmarking time series forecasting models based on the [Hugging Face `datasets`](https://huggingface.co/docs/datasets/en/index) library.
|
|
21
22
|
- **26 Nov 2024**: ⚡️ Chronos-Bolt models released [on HuggingFace](https://huggingface.co/collections/amazon/chronos-models-65f1791d630a8d57cb718444). Chronos-Bolt models are more accurate (5% lower error), up to 250x faster and 20x more memory efficient than the original Chronos models of the same size!
|
|
22
23
|
- **13 Mar 2024**: 🚀 Chronos [paper](https://arxiv.org/abs/2403.07815) and inference code released.
|
|
@@ -36,6 +37,8 @@ This package provides an interface to the Chronos family of **pretrained time se
|
|
|
36
37
|
| Model ID | Parameters |
|
|
37
38
|
| ---------------------------------------------------------------------- | ---------- |
|
|
38
39
|
| [`amazon/chronos-2`](https://huggingface.co/amazon/chronos-2) | 120M |
|
|
40
|
+
| [`autogluon/chronos-2-synth`](https://huggingface.co/autogluon/chronos-2-synth) | 120M |
|
|
41
|
+
| [`autogluon/chronos-2-small`](https://huggingface.co/autogluon/chronos-2-small) | 28M |
|
|
39
42
|
| [`amazon/chronos-bolt-tiny`](https://huggingface.co/amazon/chronos-bolt-tiny) | 9M |
|
|
40
43
|
| [`amazon/chronos-bolt-mini`](https://huggingface.co/amazon/chronos-bolt-mini) | 21M |
|
|
41
44
|
| [`amazon/chronos-bolt-small`](https://huggingface.co/amazon/chronos-bolt-small) | 48M |
|
|
@@ -57,7 +60,9 @@ pip install chronos-forecasting
|
|
|
57
60
|
```
|
|
58
61
|
|
|
59
62
|
> [!TIP]
|
|
60
|
-
> For
|
|
63
|
+
> For production use, we recommend deploying Chronos-2 to Amazon SageMaker. Two options:
|
|
64
|
+
> - **[AutoGluon-Cloud](https://auto.gluon.ai/cloud/stable/tutorials/foundation-model-timeseries.html)** (recommended) — high-level Python API. Pandas DataFrames in, forecasts out. Real-time, serverless, and batch inference out of the box.
|
|
65
|
+
> - **[SageMaker JumpStart](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb)** — production-ready real-time endpoints on CPU or GPU, ready to integrate into your existing AWS workflows.
|
|
61
66
|
|
|
62
67
|
|
|
63
68
|
### Forecasting
|
|
@@ -123,7 +128,8 @@ plt.legend()
|
|
|
123
128
|
<a href="https://colab.research.google.com/github/amazon-science/chronos-forecasting/blob/main/notebooks/chronos-2-quickstart.ipynb">
|
|
124
129
|
<img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab" height="18" align="absmiddle">
|
|
125
130
|
</a>
|
|
126
|
-
- [Deploy Chronos-2
|
|
131
|
+
- [Deploy Chronos-2 to AWS with AutoGluon-Cloud](https://auto.gluon.ai/cloud/stable/tutorials/foundation-model-timeseries.html) (recommended)
|
|
132
|
+
- [Deploy Chronos-2 to AWS with SageMaker JumpStart](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb)
|
|
127
133
|
|
|
128
134
|
## 📝 Citation
|
|
129
135
|
|