├── .github └── workflows │ ├── main.yml │ └── manual_publish.yml ├── .gitignore ├── LICENSE ├── README.md ├── pyproject.toml ├── requirements.txt ├── src └── timesfm │ ├── __init__.py │ ├── configs.py │ ├── flax │ ├── __init__.py │ ├── dense.py │ ├── normalization.py │ ├── transformer.py │ └── util.py │ ├── timesfm_2p5 │ ├── timesfm_2p5_base.py │ ├── timesfm_2p5_flax.py │ └── timesfm_2p5_torch.py │ ├── torch │ ├── __init__.py │ ├── dense.py │ ├── normalization.py │ ├── transformer.py │ └── util.py │ └── utils │ └── xreg_lib.py └── v1 ├── LICENSE ├── README.md ├── TROUBLESHOOTING.md ├── docs └── contributing.md ├── experiments ├── baselines │ ├── __init__.py │ └── timegpt_pipeline.py ├── extended_benchmarks │ ├── README.md │ ├── run_timegpt.py │ ├── run_timesfm.py │ ├── tfm_extended_new.png │ ├── tfm_results.png │ └── utils.py └── long_horizon_benchmarks │ ├── README.md │ ├── run_eval.py │ └── tfm_long_horizon.png ├── notebooks ├── covariates.ipynb ├── finetuning.ipynb └── finetuning_torch.ipynb ├── peft ├── README.md ├── finetune.py ├── finetune.sh └── usage.ipynb ├── poetry.lock ├── pyproject.toml ├── src ├── adapter │ ├── __init__.py │ ├── dora_layers.py │ ├── lora_layers.py │ └── utils.py ├── finetuning │ ├── __init__.py │ ├── finetuning_example.py │ └── finetuning_torch.py └── timesfm │ ├── __init__.py │ ├── data_loader.py │ ├── patched_decoder.py │ ├── pytorch_patched_decoder.py │ ├── time_features.py │ ├── timesfm_base.py │ ├── timesfm_jax.py │ ├── timesfm_torch.py │ └── xreg_lib.py └── tests └── test_timesfm.py /.github/workflows/main.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/.github/workflows/main.yml -------------------------------------------------------------------------------- /.github/workflows/manual_publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/.github/workflows/manual_publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/README.md -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/timesfm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/__init__.py -------------------------------------------------------------------------------- /src/timesfm/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/configs.py -------------------------------------------------------------------------------- /src/timesfm/flax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/flax/__init__.py -------------------------------------------------------------------------------- /src/timesfm/flax/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/flax/dense.py -------------------------------------------------------------------------------- /src/timesfm/flax/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/flax/normalization.py -------------------------------------------------------------------------------- /src/timesfm/flax/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/flax/transformer.py -------------------------------------------------------------------------------- /src/timesfm/flax/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/flax/util.py -------------------------------------------------------------------------------- /src/timesfm/timesfm_2p5/timesfm_2p5_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/timesfm_2p5/timesfm_2p5_base.py -------------------------------------------------------------------------------- /src/timesfm/timesfm_2p5/timesfm_2p5_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/timesfm_2p5/timesfm_2p5_flax.py -------------------------------------------------------------------------------- /src/timesfm/timesfm_2p5/timesfm_2p5_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/timesfm_2p5/timesfm_2p5_torch.py -------------------------------------------------------------------------------- /src/timesfm/torch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/torch/__init__.py -------------------------------------------------------------------------------- /src/timesfm/torch/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/torch/dense.py -------------------------------------------------------------------------------- /src/timesfm/torch/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/torch/normalization.py -------------------------------------------------------------------------------- /src/timesfm/torch/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/torch/transformer.py -------------------------------------------------------------------------------- /src/timesfm/torch/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/torch/util.py -------------------------------------------------------------------------------- /src/timesfm/utils/xreg_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/src/timesfm/utils/xreg_lib.py -------------------------------------------------------------------------------- /v1/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/LICENSE -------------------------------------------------------------------------------- /v1/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/README.md -------------------------------------------------------------------------------- /v1/TROUBLESHOOTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/TROUBLESHOOTING.md -------------------------------------------------------------------------------- /v1/docs/contributing.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/docs/contributing.md -------------------------------------------------------------------------------- /v1/experiments/baselines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/experiments/baselines/__init__.py -------------------------------------------------------------------------------- /v1/experiments/baselines/timegpt_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/experiments/baselines/timegpt_pipeline.py -------------------------------------------------------------------------------- /v1/experiments/extended_benchmarks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/experiments/extended_benchmarks/README.md -------------------------------------------------------------------------------- /v1/experiments/extended_benchmarks/run_timegpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/experiments/extended_benchmarks/run_timegpt.py -------------------------------------------------------------------------------- /v1/experiments/extended_benchmarks/run_timesfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/experiments/extended_benchmarks/run_timesfm.py -------------------------------------------------------------------------------- /v1/experiments/extended_benchmarks/tfm_extended_new.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/experiments/extended_benchmarks/tfm_extended_new.png -------------------------------------------------------------------------------- /v1/experiments/extended_benchmarks/tfm_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/experiments/extended_benchmarks/tfm_results.png -------------------------------------------------------------------------------- /v1/experiments/extended_benchmarks/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/experiments/extended_benchmarks/utils.py -------------------------------------------------------------------------------- /v1/experiments/long_horizon_benchmarks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/experiments/long_horizon_benchmarks/README.md -------------------------------------------------------------------------------- /v1/experiments/long_horizon_benchmarks/run_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/experiments/long_horizon_benchmarks/run_eval.py -------------------------------------------------------------------------------- /v1/experiments/long_horizon_benchmarks/tfm_long_horizon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/experiments/long_horizon_benchmarks/tfm_long_horizon.png -------------------------------------------------------------------------------- /v1/notebooks/covariates.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/notebooks/covariates.ipynb -------------------------------------------------------------------------------- /v1/notebooks/finetuning.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/notebooks/finetuning.ipynb -------------------------------------------------------------------------------- /v1/notebooks/finetuning_torch.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/notebooks/finetuning_torch.ipynb -------------------------------------------------------------------------------- /v1/peft/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/peft/README.md -------------------------------------------------------------------------------- /v1/peft/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/peft/finetune.py -------------------------------------------------------------------------------- /v1/peft/finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/peft/finetune.sh -------------------------------------------------------------------------------- /v1/peft/usage.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/peft/usage.ipynb -------------------------------------------------------------------------------- /v1/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/poetry.lock -------------------------------------------------------------------------------- /v1/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/pyproject.toml -------------------------------------------------------------------------------- /v1/src/adapter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/adapter/__init__.py -------------------------------------------------------------------------------- /v1/src/adapter/dora_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/adapter/dora_layers.py -------------------------------------------------------------------------------- /v1/src/adapter/lora_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/adapter/lora_layers.py -------------------------------------------------------------------------------- /v1/src/adapter/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/adapter/utils.py -------------------------------------------------------------------------------- /v1/src/finetuning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /v1/src/finetuning/finetuning_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/finetuning/finetuning_example.py -------------------------------------------------------------------------------- /v1/src/finetuning/finetuning_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/finetuning/finetuning_torch.py -------------------------------------------------------------------------------- /v1/src/timesfm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/timesfm/__init__.py -------------------------------------------------------------------------------- /v1/src/timesfm/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/timesfm/data_loader.py -------------------------------------------------------------------------------- /v1/src/timesfm/patched_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/timesfm/patched_decoder.py -------------------------------------------------------------------------------- /v1/src/timesfm/pytorch_patched_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/timesfm/pytorch_patched_decoder.py -------------------------------------------------------------------------------- /v1/src/timesfm/time_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/timesfm/time_features.py -------------------------------------------------------------------------------- /v1/src/timesfm/timesfm_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/timesfm/timesfm_base.py -------------------------------------------------------------------------------- /v1/src/timesfm/timesfm_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/timesfm/timesfm_jax.py -------------------------------------------------------------------------------- /v1/src/timesfm/timesfm_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/timesfm/timesfm_torch.py -------------------------------------------------------------------------------- /v1/src/timesfm/xreg_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/src/timesfm/xreg_lib.py -------------------------------------------------------------------------------- /v1/tests/test_timesfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/timesfm/HEAD/v1/tests/test_timesfm.py --------------------------------------------------------------------------------