🚀 What's New?

• Chronos-2 can now be deployed on AWS via SageMaker JumpStart. Please check this notebook for details.
• Scaled Dot Product Attention (SDPA) is now used as the default attention implementation in Chronos-2. If you need to use the previous eager implementation, please load the model with Chronos2Pipeline.from_pretrained(..., attn_implementation="eager").
• predict_df support has been added for older Chronos and Chronos-Bolt models. Now, all models (Chronos-2, Chronos-Bolt, Chronos) provide a unified pandas dataframe API. Note: Only Chronos-2 supports multivariate and covariate-informed forecasting.
• Chronos2Pipeline.embed has been added, enabling users to extract embeddings from the last layer of the Chronos-2 encoder.

🐛 Bug Fixes

• Fixed issues related to past-only covariates use during Chronos-2 fine-tuning. If you're fine-tuning Chronos-2 models, we strongly recommend upgrading to chronos-forecasting==2.1.0.
• Fixed issue related to multiple workers on windows.

All Changes

• Fix pandas install instruction by @abdulfatir in Fix pandas install instruction  #326
• Scope down GitHub token permissions by @AdnaneKhan in Scope down GitHub token permissions #328
• Update Chronos-2 notebook install instructions by @abdulfatir in Update Chronos-2 notebook install instructions #333
• Add logging steps in fine-tuning example by @abdulfatir in Add logging steps in fine-tuning example #334
• [chronos-2] add support for SDPA by @kashif in [chronos-2] add support for SDPA #331
• Treat notebooks as documentation in linguist by @abdulfatir in Treat notebooks as documentation in linguist #338
• Count notebooks as python in language stats by @abdulfatir in Count notebooks as python in language stats #347
• Move SageMaker notebook to a new path by @shchur in Move SageMaker notebook to a new path #348
• Revert "Move SageMaker notebook to a new path" by @shchur in Revert "Move SageMaker notebook to a new path" #349
• fix validate_and_prepare_single_dict_task by separating "past only" and "future known" cov keys by @HarvestStars in fix validate_and_prepare_single_dict_task by separating "past only" and "future known" cov keys #344
• Add Chronos-2 SageMaker notebook by @shchur in Add Chronos-2 SageMaker notebook #350
• Update model link to Hugging Face by @shchur in Update model link to Hugging Face #353
• Remove logo from README by @shchur in Remove logo from README #355
• Update Notebooks and README by @abdulfatir in Update Notebooks and README #356
• Add FutureWarning for CloudFront by @abdulfatir in Add FutureWarning for CloudFront #357
• Relax transformers lower bound to >=4.41 by @abdulfatir in Relax transformers lower bound to >=4.41 #364
• Set num_workers=0 in Chronos-2 test data loader by @IliasAarab in Set num_workers=0 in Chronos-2 test data loader #365
• Update long horizon heuristic for Chronos-Bolt by @abdulfatir in Update long horizon heuristic for Chronos-Bolt #366
• Add predict_df support for Chronos and Chronos-Bolt by @abdulfatir in Add predict_df support for Chronos and Chronos-Bolt #371
• Add Chronos2Pipeline.embed by @abdulfatir in Add Chronos2Pipeline.embed #361
• Mask past-only covariates during loss computation by @abdulfatir in Mask past-only covariates during loss computation #379
• Chronos-2: Change default fine-tuning learning rate and remove experimental label by @abdulfatir in Chronos-2: Change default fine-tuning learning rate and remove experimental label #381
• Update version to 2.1.0rc1 for pre-release by @abdulfatir in Update version to 2.1.0rc1 for pre-release #383
• Update torch dependency version to >=2.2,<3 by @abdulfatir in Update torch dependency version to >=2.2,<3 #388
• Bump version to 2.1.0 by @abdulfatir in Bump version to 2.1.0 #389

New Contributors

• @AdnaneKhan made their first contribution in Scope down GitHub token permissions #328
• @kashif made their first contribution in [chronos-2] add support for SDPA #331
• @HarvestStars made their first contribution in fix validate_and_prepare_single_dict_task by separating "past only" and "future known" cov keys #344
• @IliasAarab made their first contribution in Set num_workers=0 in Chronos-2 test data loader #365

Full Changelog: v2.0.1...v2.1.0