Commit Graph

196 Commits

Author SHA1 Message Date
misha-chertushkin 67e0eaaa3d Quantiles PR nit fix 2025-02-05 13:41:13 +00:00
misha-chertushkin 402ebf52a5 Small fix with default value 2025-02-01 02:43:56 +00:00
misha-chertushkin ca87a438a0 Update examples with new feedback 2025-02-01 02:30:17 +00:00
misha-chertushkin bca190dae6 PR Feedback 2025-02-01 02:26:39 +00:00
misha-chertushkin 86551f761a Add jupyter notebook 2025-01-21 19:57:50 +00:00
misha-chertushkin f84366e3d1 Style fix 2025-01-21 19:29:42 +00:00
misha-chertushkin 4f4649c212 Small refactoring 2025-01-21 16:09:28 +00:00
misha-chertushkin 955d6cda22 Gpu support finished 2025-01-21 15:52:08 +00:00
misha-chertushkin a559718c66 GPU support added, almost done 2025-01-21 02:19:35 +00:00
misha-chertushkin fb6213ed59 Fix Wandb errro 2025-01-21 01:50:18 +00:00
misha-chertushkin 4942cda83b Refactor into 2 examples 2025-01-21 01:45:30 +00:00
misha-chertushkin 65700643df Add finetuning support 2025-01-20 21:48:53 +00:00
Rajat Sen ded73ed55f Merge pull request #218 from google-research/rajat_dev
Removing ipython dependency
2025-01-15 14:11:58 -08:00
Rajat Sen 13ac0c6435 poetry push version 2025-01-15 22:11:12 +00:00
Rajat Sen a9c5b48c49 Merge branch 'master' of https://github.com/google-research/timesfm into rajat_dev 2025-01-15 21:24:14 +00:00
Rajat Sen c49cf9caf8 remove ipython dep 2025-01-15 21:21:54 +00:00
Rajat Sen a3ab7873e9 Merge pull request #213 from google-research/siriuz42-patch-1
Update timesfm_base.py
2025-01-10 16:56:51 -08:00
Yichen Zhou 95db25be3e Update timesfm_base.py
Fixing MS checked after S
2025-01-10 15:26:01 -08:00
Rajat Sen e628f0fec1 Merge pull request #210 from google-research/rajat_dev
updating benchmarks to point to v2.0
2025-01-08 10:52:33 -08:00
Rajat Sen 8fe4a0a511 updating benchmarks to point to v2.0 2025-01-08 18:51:47 +00:00
Rajat Sen ff2a350ba3 Merge pull request #208 from google-research/rajat_dev
minor update to readme
2025-01-02 17:26:27 -08:00
Rajat Sen 1e249ef0b1 minor update to readme 2025-01-02 23:56:33 +00:00
Rajat Sen e77303c551 Merge pull request #205 from google-research/rajat_dev
minor changes to readme
2024-12-30 16:21:00 -08:00
Rajat Sen 5a62d02caf minor changes to readme 2024-12-31 00:19:59 +00:00
Rajat Sen 028188be4d Merge pull request #204 from google-research/rajat_dev
Adding v2.0 support
2024-12-30 16:09:13 -08:00
Rajat Sen 4aec27ee02 push poetry version 2024-12-31 00:07:59 +00:00
Rajat Sen db5fb13c62 minor changes to readme links 2024-12-30 23:59:37 +00:00
Rajat Sen 73704e5936 Adding v2.0 support 2024-12-30 23:55:36 +00:00
Rajat Sen 5a69171296 Merge pull request #202 from google-research/rajat_dev
bump version number
2024-12-23 16:24:30 -08:00
Rajat Sen 8fc2f2f478 bump version number 2024-12-24 00:23:27 +00:00
Yichen Zhou 154248137c Merge pull request #198 from google-research/timesfm_2d0
Nan handling
2024-12-20 15:43:30 -08:00
Yichen Zhou faf1c7681e nan handling 2024-12-20 14:18:24 -08:00
Yichen Zhou 562187c6d1 nan handling 2024-12-20 14:01:06 -08:00
Yichen Zhou 27f40371c2 add support for inference sans positional encoding 2024-12-12 11:04:21 -08:00
Rajat Sen 02bc2f2212 Merge pull request #182 from google-research/rajat_dev
Fix median index issue for jax
2024-11-09 12:29:19 -08:00
Rajat Sen eb02472b88 nump version 2024-11-09 20:28:42 +00:00
Rajat Sen c0709fdd69 fix median index issue 2024-11-09 20:28:02 +00:00
Yichen Zhou 0706b0fb5c Merge pull request #178 from google-research/rajat_dev
Setting median outputs as default. Also minor changes to finetuning.
2024-11-05 16:31:23 -08:00
Yichen Zhou 9b302ae794 Merge pull request #177 from andreramosfdc/allow_local_dir_download
Enable local model snapshot downloads
2024-11-05 16:28:19 -08:00
Rajat Sen be37abadfc push poetry build version up 2024-11-05 22:55:20 +00:00
Rajat Sen 76141bf917 Setting median outputs as default. Also minor changes to finetuning notebook. 2024-11-05 22:51:07 +00:00
Rajat Sen b0cbbbfc1e Merge branch 'master' of https://github.com/google-research/timesfm into rajat_dev 2024-11-05 22:28:21 +00:00
andreramosfdc c1e2094a10 Enable model snapshot downloads to be saved in a specified local directory. 2024-10-30 18:38:11 -03:00
Rajat Sen e3d45d6568 Merge pull request #161 from google-research/rajat_dev
updating readme with minor changes to installation instructions
2024-10-17 16:11:04 -07:00
Rajat Sen e2f2e68de5 Merge branch 'master' of https://github.com/google-research/timesfm into rajat_dev 2024-10-17 23:09:57 +00:00
Rajat Sen a3a1c3b271 updating readme with minor changes to installation instructions 2024-10-17 23:09:04 +00:00
Rajat Sen e4a2f175a9 Merge pull request #160 from google-research/rajat_dev
Fixes build and upload to pypi
2024-10-17 15:57:06 -07:00
Rajat Sen 6a22971c78 updating poetry version 2024-10-17 22:55:59 +00:00
Rajat Sen 18f2314da4 fixes build push 2024-10-17 22:54:11 +00:00
Rajat Sen 280b16ee89 pypi package for both torch and pax versions 2024-10-17 22:51:29 +00:00