diff --git a/README.md b/README.md index 09685f42..94fa5335 100644 --- a/README.md +++ b/README.md @@ -56,6 +56,10 @@ To perform inference with Chronos, the easiest way is to install this package th pip install chronos-forecasting ``` +> [!TIP] +> 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. + + ### Forecasting A minimal example showing how to perform forecasting using Chronos-2: @@ -111,8 +115,15 @@ plt.legend() ## Example Notebooks - [Chronos-2 Quick Start](notebooks/chronos-2-quickstart.ipynb) -- [Deploy Chronos-Bolt on Amazon SageMaker](notebooks/deploy-chronos-bolt-to-amazon-sagemaker.ipynb) -- Deploy Chronos-2 on Amazon SageMaker (coming soon!) +   + + Open In SageMaker Studio Lab + +   + + Open In Colab + +- [Deploy Chronos-2 on Amazon SageMaker](notebooks/deploy-chronos-to-amazon-sagemaker.ipynb) ## 📝 Citation diff --git a/notebooks/chronos-2-quickstart.ipynb b/notebooks/chronos-2-quickstart.ipynb index 32a6111c..65ae9aa7 100644 --- a/notebooks/chronos-2-quickstart.ipynb +++ b/notebooks/chronos-2-quickstart.ipynb @@ -7,6 +7,12 @@ "source": [ "# Getting Started with Chronos-2\n", "\n", + "[![Open In SageMaker Studio Lab](https://studiolab.sagemaker.aws/studiolab.svg)](https://studiolab.sagemaker.aws/import/github/amazon-science/chronos-forecasting/blob/main/notebooks/chronos-2-quickstart.ipynb)\n", + "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](\n", + "https://colab.research.google.com/github/amazon-science/chronos-forecasting/blob/main/notebooks/chronos-2-quickstart.ipynb)\n", + "\n", + "\n", + "\n", "**Chronos-2** is a foundation model for time series forecasting that builds on [Chronos](https://arxiv.org/abs/2403.07815) and [Chronos-Bolt](https://aws.amazon.com/blogs/machine-learning/fast-and-accurate-zero-shot-forecasting-with-chronos-bolt-and-autogluon/). It offers significant improvements in capabilities and can handle diverse forecasting scenarios not supported by earlier models.\n", "\n", "| Capability | Chronos | Chronos-Bolt | Chronos-2 |\n", @@ -31,7 +37,7 @@ "metadata": {}, "outputs": [], "source": [ - "%pip install -U 'chronos-forecasting>=2.0' 'pandas[pyarrow]' 'matplotlib'" + "%pip install 'chronos-forecasting>=2.0' 'pandas[pyarrow]' 'matplotlib'" ] }, { @@ -41,9 +47,9 @@ "metadata": {}, "outputs": [], "source": [ - "# Use only 1 GPU if available\n", "import os\n", "\n", + "# Use only 1 GPU if available\n", "os.environ[\"CUDA_VISIBLE_DEVICES\"] = \"0\"\n", "\n", "import pandas as pd\n", @@ -1586,7 +1592,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.11.11" + "version": "3.11.13" } }, "nbformat": 4,