v1.2.6

yarnpkg/yarnv1.2.6Dec 31, 2024by rajatsen91

AI Summary

This release adds support for TimesFM-2.0 models, sets the median head as the default point forecaster, and includes bug fixes for PyTorch decoding.

Key Highlights

Added support for TimesFM-2.0 models (500m variants for JAX and PyTorch)
Median head set as the default point forecaster
Added hyperparameter support for TimesFM-2.0
Bug fixes in PyTorch decoding
Cached decoding support removed for both JAX and PyTorch

Breaking Changes

Cached decoding support has been removed for both JAX and PyTorch backends

New Features

TimesFM-2.0 model support
Hyperparameter support for TimesFM-2.0
Default point forecaster changed to median head

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