Merge branch 'master' of github.com:google-research/timesfm

pyproject.toml

@@ -1,10 +1,25 @@
-[tool.poetry]
-name = "timesfm"
+# This project can be installed with `python3 -m pip install -e .` from the main directory. 
+
+[project]
+name = "timesfm-jax"
 packages = [
     { include = "*", from = "src" },
 ]
 version = "0.0.1"
 description = "Open weights time-series foundation model from Google Research."
+version = "1.0.1"
+dependencies = [
+    "jax==0.4.26",
+    "paxml==1.4.0",
+    "praxis==1.4.0",
+    "jaxlib==0.4.26",
+    "numpy==1.26.4",
+    "pandas==2.1.4",
+    "einshape==1.0.0",
+    "utilsforecast==0.1.10",
+    "huggingface_hub[cli]==0.23.0", 
+    "scikit-learn==1.5.1",
+]
 authors = [
     "Rajat Sen <senrajat@google.com>",
     "Yichen Zhou <yichenzhou@google.com>",