v1.2.6

usestrix/strixv1.2.6Dec 31, 2024by rajatsen91

AI Summary

This release introduces support for TimesFM-2.0 models, updates the default point forecaster to the median head, and includes bug fixes for PyTorch decoding.

Key Highlights

Added support for TimesFM-2.0 models.
Set the median head as the default point forecaster.
Added hyperparameter support for TimesFM-2.0.
Fixed some bugs in PyTorch decoding.

New Features

TimesFM-2.0 model support
Hyperparameter support for TimesFM-2.0
Fixed PyTorch decoding bugs

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