Justin Guese
60 lines
1.5 KiB
TOML
60 lines
1.5 KiB
TOML
# 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>",
|
|
"Abhimanyu Das <abhidas@google.com>",
|
|
"Petros Mol <pmol@google.com>",
|
|
"Justin Güse <guese.justin@gmail.com>",
|
|
]
|
|
readme = "README.md"
|
|
keywords = ["time series", "timesfm", "forecast", "time series model"]
|
|
homepage = "https://github.com/google-research/timesfm"
|
|
repository = "https://github.com/google-research/timesfm"
|
|
classifiers = [
|
|
"Environment :: Console",
|
|
"Framework :: Flake8",
|
|
"Operating System :: OS Independent",
|
|
"Topic :: Software Development :: Documentation",
|
|
"Topic :: Software Development :: Libraries :: Python Modules",
|
|
"Topic :: Software Development :: Quality Assurance",
|
|
]
|
|
include = [
|
|
"LICENSE",
|
|
]
|
|
|
|
[tool.poetry.dependencies]
|
|
python = ">=3.10,<3.11"
|
|
einshape = ">=1.0.0"
|
|
numpy = ">=1.26.4"
|
|
pandas = ">=2.1.4"
|
|
paxml = "1.4.0"
|
|
jax = "0.4.26"
|
|
jaxlib = "0.4.26"
|
|
utilsforecast = "^0.1.10"
|
|
|
|
|
|
[build-system]
|
|
requires = ["poetry-core"]
|
|
build-backend = "poetry.core.masonry.api"
|