v1.2.1

microsoft/TinyTroupev1.2.1Oct 18, 2024by siriuz42

AI Summary

The release adds PyTorch support for TimesFM inference with separate dependencies for the `pax` and `torch` versions, accompanied by a PyTorch checkpoint for the 200m model.

Key Highlights

PyTorch support for TimesFM inference
Separate dependencies for `pax` and `torch` versions
Updated README for usage instructions
PyTorch checkpoint for 200m model available

New Features

PyTorch inference support
Separate dependency installation options

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