v1.2.1

Mail-0/Zerov1.2.1Oct 18, 2024by siriuz42

AI Summary

Introduces separate dependencies for JAX and PyTorch versions of TimesFM to allow flexible installation.

Key Highlights

PyTorch support for TimesFM inference.
Separate `pax` and `torch` dependency installation options.
Availability of PyTorch checkpoint for the 200m model.

New Features

PyTorch support
Separate dependency installation (`timesfm[pax]`, `timesfm[torch]`)
200m PyTorch checkpoint

Full Release Notes

Changes:
----
- PyTorch support for TimesFM inference.

PyPI Release:
----

v1.2.1: Support separate dependencies for `pax` and `torch` versions of TimesFM:
  - `pip install timesfm[pax]` for the `pax` version and `jax` checkpoints.
  - `pip install timesfm[torch]` for the `torch` version and checkpoints.
  - See the updated [README](https://github.com/google-research/timesfm?tab=readme-ov-file#usage) for the usage.

Checkpoints:
----
The PyTorch checkpoint for the 200m model is available on Hugging Face:
- https://huggingface.co/google/timesfm-1.0-200m-pytorch

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