Yichen Zhou
30 lines
823 B
TOML
30 lines
823 B
TOML
# This project can be installed with `python3 -m pip install -e .` from the main directory.
|
|
|
|
[project]
|
|
name = "timesfm-jax"
|
|
description = "Open weights time-series foundation model from Google Research."
|
|
version = "1.0.1"
|
|
dependencies = [
|
|
"jax[cuda12]==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.0.2",
|
|
]
|
|
authors = [
|
|
{name = "Rajat Sen", email = "senrajat@google.com"},
|
|
{name = "Yichen Zhou", email = "yichenzhou@google.com"},
|
|
{name = "Abhimanyu Das", email = "abhidas@google.com"},
|
|
{name = "Petros Mol", email = "pmol@google.com"},
|
|
]
|
|
|
|
|
|
[build-system]
|
|
requires = ["setuptools"]
|
|
build-backend = "setuptools.build_meta"
|