Commit Graph

204 Commits

Author SHA1 Message Date
misha-chertushkin b45e8d69e6 Add finetuning folder 2025-03-05 21:08:22 +00:00
Rajat Sen 057c9992bd Merge pull request #243 from misha-chertushkin/notebook-import-fix
Fix imports in Notebook examples
2025-02-17 09:34:25 -08:00
misha-chertushkin cb578f30c5 Update readme 2025-02-17 12:00:20 +00:00
misha-chertushkin 5117a0e338 Fix the data function 2025-02-17 11:51:20 +00:00
misha-chertushkin b3052cae9d Fix import in notebook 2025-02-17 11:39:33 +00:00
Rajat Sen c9a582506b Merge pull request #223 from misha-chertushkin/master
Finetuning + Multi-GPU Support with PyTorch
2025-02-05 16:27:08 -08:00
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
Rajat Sen 9594c0618d Merge pull request #227 from google-research/rajat_dev
Revising messages during import
2025-01-22 14:01:56 -08:00
Rajat Sen 1f6146188c Revising messages during import 2025-01-22 21:58:28 +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