v1.2.1

BayramAnnakov/ux-waiting-auditv1.2.1Oct 18, 2024by siriuz42

AI Summary

The release adds full PyTorch support for TimesFM inference, allowing users to run models with PyTorch checkpoints. It also introduces separate dependency installation options for JAX and PyTorch versions.

Key Highlights

PyTorch support for TimesFM inference.
Separate dependencies for `pax` and `torch` versions.
PyTorch checkpoint for the 200m model available on Hugging Face.
Updated README with usage instructions.

New Features

PyTorch inference support
Separate pip install options (`timesfm[pax]` and `timesfm[torch]`)

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