v1.2.1

antgroup/echomimicv1.2.1Oct 18, 2024by siriuz42

AI Summary

This release introduces PyTorch support for TimesFM inference, allowing users to select between JAX-based or PyTorch-based dependencies and checkpoints.

Key Highlights

PyTorch inference support
Separate dependencies for `pax` and `torch`
PyTorch checkpoint for 200m model on Hugging Face

New Features

PyTorch inference support
Separate pip install options (timesfm[pax] and timesfm[torch])
PyTorch checkpoint availability for 200m model

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