v1.2.6

rohitg00/agentmemoryv1.2.6Dec 31, 2024by rajatsen91

AI Summary

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

Key Highlights

Support for TimesFM-2.0 models (JAX and PyTorch variants)
Median head set as the default point forecaster
Hyperparameter support added for TimesFM-2.0 models
Bug fixes in PyTorch decoding implementation

New Features

TimesFM-2.0 model support (500m-jax and 500m-pytorch checkpoints)
Hyperparameter support for TimesFM-2.0 models
Updated default point forecaster to median head
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