v1.2.6

tracel-ai/burnv1.2.6Dec 31, 2024by rajatsen91

AI Summary

This release introduces support for TimesFM-2.0 models and sets the median head as the default point forecaster. It includes hyperparameter support and fixes for PyTorch decoding, while noting that cached decoding is currently unsupported.

Key Highlights

Add support for TimesFM-2.0 models
Set median head as the default point forecaster
Add hyperparameter support for TimesFM-2.0
Fixes in PyTorch decoding

New Features

TimesFM-2.0 model support
Median head as default point forecaster
Hyperparameter support for TimesFM-2.0
PyTorch decoding bug fixes

Full Release Notes

Changes:
----
1. Add support for TimesFM-2.0 models.
2. Set the median head as the default point forecaster.

PyPI Release:
----
v1.2.6: Support for TimesFM-2.0 models.
- Add hparam support for TimesFM-2.0 models.
- Some bug fixes in pytorch decoding.
- Right now we do not support cached decoding in both jax and pytorch.

Checkpoints:
----
The TimesFM-2.0 checkpoints are available on Hugging Face:
- https://huggingface.co/google/timesfm-2.0-500m-jax
- https://huggingface.co/google/timesfm-2.0-500m-pytorch

Full Changelog:
----
https://github.com/google-research/timesfm/commits/v1.2.6