v1.2.1

abhigyanpatwari/GitNexusv1.2.1Oct 18, 2024by siriuz42

AI Summary

Introduces PyTorch support for TimesFM inference and separates dependencies into `timesfm[pax]` (JAX) and `timesfm[torch]` packages.

Key Highlights

PyTorch inference support
Separate dependency installation options
PyTorch checkpoint availability on Hugging Face

New Features

PyTorch inference engine
JAX dependency installation
Torch dependency installation

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