v1.2.1

tambo-ai/tambov1.2.1Oct 18, 2024by siriuz42

AI Summary

Introduces PyTorch support for inference with separate dependency management via pip install options, alongside the availability of a PyTorch checkpoint.

Key Highlights

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

New Features

PyTorch inference
Separate dependency management
New checkpoint availability

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