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.
Files changed (99) hide show
  1. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.github/ISSUE_TEMPLATE/bug_report.md +1 -1
  2. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/PKG-INFO +20 -16
  3. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/README.md +9 -3
  4. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/notebooks/chronos-2-quickstart.ipynb +105 -58
  5. chronos_forecasting-2.3.1/notebooks/deploy-chronos-to-amazon-sagemaker.ipynb +1435 -0
  6. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/pyproject.toml +10 -12
  7. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/train.py +3 -1
  8. chronos_forecasting-2.3.1/src/chronos/__about__.py +1 -0
  9. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/base.py +122 -98
  10. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/boto_utils.py +12 -5
  11. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos2/config.py +4 -1
  12. chronos_forecasting-2.3.1/src/chronos/chronos2/dataset.py +373 -0
  13. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos2/layers.py +62 -17
  14. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos2/model.py +37 -18
  15. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos2/pipeline.py +152 -177
  16. chronos_forecasting-2.3.1/src/chronos/chronos2/preprocess.py +686 -0
  17. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos_bolt.py +48 -11
  18. chronos_forecasting-2.3.1/src/chronos/df_utils.py +338 -0
  19. chronos_forecasting-2.3.1/test/dummy-chronos-model/model.safetensors +0 -0
  20. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/test_chronos2.py +226 -58
  21. chronos_forecasting-2.3.1/test/test_df_utils.py +369 -0
  22. chronos_forecasting-2.3.1/test/test_preprocess.py +575 -0
  23. chronos_forecasting-2.2.2/notebooks/deploy-chronos-to-amazon-sagemaker.ipynb +0 -1173
  24. chronos_forecasting-2.2.2/src/chronos/__about__.py +0 -1
  25. chronos_forecasting-2.2.2/src/chronos/chronos2/dataset.py +0 -655
  26. chronos_forecasting-2.2.2/src/chronos/df_utils.py +0 -323
  27. chronos_forecasting-2.2.2/test/dummy-chronos-model/pytorch_model.bin +0 -0
  28. chronos_forecasting-2.2.2/test/test_df_utils.py +0 -361
  29. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.gitattributes +0 -0
  30. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.github/ISSUE_TEMPLATE/config.yml +0 -0
  31. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.github/workflows/ci.yml +0 -0
  32. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.github/workflows/eval-model.yml +0 -0
  33. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.github/workflows/publish-to-pypi.yml +0 -0
  34. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/.gitignore +0 -0
  35. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/CITATION.cff +0 -0
  36. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/CODE_OF_CONDUCT.md +0 -0
  37. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/CONTRIBUTING.md +0 -0
  38. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/LICENSE +0 -0
  39. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/NOTICE +0 -0
  40. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/ci/evaluate/backtest_config.yaml +0 -0
  41. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/README.md +0 -0
  42. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/agg-relative-score.py +0 -0
  43. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/configs/in-domain.yaml +0 -0
  44. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/configs/zero-shot.yaml +0 -0
  45. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/evaluate.py +0 -0
  46. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-base-agg-rel-scores.csv +0 -0
  47. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-base-in-domain.csv +0 -0
  48. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-base-zero-shot.csv +0 -0
  49. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-mini-agg-rel-scores.csv +0 -0
  50. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-mini-in-domain.csv +0 -0
  51. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-mini-zero-shot.csv +0 -0
  52. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-small-agg-rel-scores.csv +0 -0
  53. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-small-in-domain.csv +0 -0
  54. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-small-zero-shot.csv +0 -0
  55. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-tiny-agg-rel-scores.csv +0 -0
  56. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-tiny-in-domain.csv +0 -0
  57. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-bolt-tiny-zero-shot.csv +0 -0
  58. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-base-agg-rel-scores.csv +0 -0
  59. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-base-in-domain.csv +0 -0
  60. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-base-zero-shot.csv +0 -0
  61. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-large-agg-rel-scores.csv +0 -0
  62. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-large-in-domain.csv +0 -0
  63. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-large-zero-shot.csv +0 -0
  64. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-mini-agg-rel-scores.csv +0 -0
  65. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-mini-in-domain.csv +0 -0
  66. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-mini-zero-shot.csv +0 -0
  67. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-small-agg-rel-scores.csv +0 -0
  68. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-small-in-domain.csv +0 -0
  69. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-small-zero-shot.csv +0 -0
  70. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-tiny-agg-rel-scores.csv +0 -0
  71. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-tiny-in-domain.csv +0 -0
  72. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/chronos-t5-tiny-zero-shot.csv +0 -0
  73. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/seasonal-naive-in-domain.csv +0 -0
  74. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/evaluation/results/seasonal-naive-zero-shot.csv +0 -0
  75. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/kernel-synth.py +0 -0
  76. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/configs/chronos-gpt2.yaml +0 -0
  77. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/configs/chronos-t5-base.yaml +0 -0
  78. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/configs/chronos-t5-large.yaml +0 -0
  79. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/configs/chronos-t5-mini.yaml +0 -0
  80. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/configs/chronos-t5-small.yaml +0 -0
  81. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/scripts/training/configs/chronos-t5-tiny.yaml +0 -0
  82. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/__init__.py +0 -0
  83. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos.py +0 -0
  84. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos2/__init__.py +0 -0
  85. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/chronos2/trainer.py +0 -0
  86. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/src/chronos/utils.py +0 -0
  87. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/__init__.py +0 -0
  88. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos-bolt-model/config.json +0 -0
  89. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos-bolt-model/model.safetensors +0 -0
  90. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos-model/config.json +0 -0
  91. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos-model/generation_config.json +0 -0
  92. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos2-lora/adapter_config.json +0 -0
  93. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos2-lora/adapter_model.safetensors +0 -0
  94. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos2-model/config.json +0 -0
  95. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/dummy-chronos2-model/model.safetensors +0 -0
  96. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/test_chronos.py +0 -0
  97. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/test_chronos_bolt.py +0 -0
  98. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/test_utils.py +0 -0
  99. {chronos_forecasting-2.2.2 → chronos_forecasting-2.3.1}/test/util.py +0 -0
@@ -30,4 +30,4 @@ CUDA version:
30
30
  PyTorch version:
31
31
  HuggingFace transformers version:
32
32
  HuggingFace accelerate version:
33
-
33
+ Pandas version:
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: chronos-forecasting
3
- Version: 2.2.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.34
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: scikit-learn<2,>=1.6.0
195
+ Requires-Dist: pandas<4,>=2.0
196
196
  Requires-Dist: torch<3,>=2.2
197
- Requires-Dist: transformers<5,>=4.41
197
+ Requires-Dist: transformers<6,>=4.41
198
198
  Provides-Extra: dev
199
- Requires-Dist: datasets<4,>=3.0; extra == 'dev'
200
- Requires-Dist: fev>=0.6.1; extra == 'dev'
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>=0.6.1; extra == 'extras'
211
- Requires-Dist: pandas[pyarrow]<2.4,>=2.0; extra == 'extras'
212
- Requires-Dist: peft<0.18,>=0.13.0; extra == 'extras'
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>=0.6.1; extra == 'test'
216
- Requires-Dist: pandas[pyarrow]<2.4,>=2.0; extra == 'test'
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 reliable production use, we recommend using Chronos-2 models through [Amazon SageMaker JumpStart](https://aws.amazon.com/sagemaker/ai/jumpstart/). Check out [this tutorial](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb) to learn how to deploy Chronos-2 inference endpoints to AWS with just a few lines of code.
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 on Amazon SageMaker](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb)
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 reliable production use, we recommend using Chronos-2 models through [Amazon SageMaker JumpStart](https://aws.amazon.com/sagemaker/ai/jumpstart/). Check out [this tutorial](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb) to learn how to deploy Chronos-2 inference endpoints to AWS with just a few lines of code.
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 on Amazon SageMaker](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb)
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