v1.2.1

tracel-ai/burnv1.2.1Oct 18, 2024by siriuz42

AI Summary

This release introduces PyTorch support for TimesFM inference, allowing users to install dependencies separately for the Pax (JAX) or Torch versions. It also provides a PyTorch checkpoint for the 200m model on Hugging Face.

Key Highlights

PyTorch support for TimesFM inference
Separate dependencies for Pax and Torch versions
PyTorch checkpoint for the 200m model

New Features

PyTorch inference support
Separate dependency installation options (pax vs torch)
PyTorch 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