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 thepax
version andjax
checkpoints.pip install timesfm[torch]
for thetorch
version and checkpoints.- See the updated README for the usage.
Checkpoints:
The PyTorch checkpoint for the 200m model is available on Hugging Face: