Commit Graph

277 Commits

Author SHA1 Message Date
Rajat Sen be91ef8ee3 manually download config.json 2025-10-30 20:11:16 +00:00
Yichen Zhou c43e8f3e66 Update README.md 2025-10-29 17:45:32 -07:00
Yichen Zhou a7435a190f Merge pull request #329 from google-research/siriuz42-2.0-pr
Bring back xreg for TimesFM 2.5
2025-10-29 17:41:32 -07:00
siriuz42 ba2417e11a update requirement 2025-10-28 22:40:21 +00:00
siriuz42 0d4d5141d7 covariate support 2025-10-28 22:30:24 +00:00
siriuz42 c24db2afd2 covariate support 2025-10-28 22:29:40 +00:00
siriuz42 80f1a1eafe Merge remote-tracking branch 'origin/master' into siriuz42-2.0-pr 2025-10-28 17:50:55 +00:00
Rajat Sen 58e01ad82f Merge pull request #326 from google-research/siriuz42-patch-1
Update timesfm_2p5_flax.py
2025-10-08 16:15:28 -07:00
Yichen Zhou a6609df1d5 Update timesfm_2p5_flax.py
bug fix
2025-10-08 16:03:18 -07:00
siriuz42 b6b207a4ed bug fix 2025-10-08 22:59:07 +00:00
Rajat Sen ce168ffd1f Merge pull request #325 from google-research/siriuz42-2.0-pr
Manually trigger jitting in Flax.
2025-10-08 09:37:10 -07:00
siriuz42 9080939cb8 Manually trigger jitting in Flax. 2025-10-08 16:32:41 +00:00
Rajat Sen a21b1d5b25 Merge pull request #324 from google-research/rajat-dev
Use uv instead of poetry
2025-10-08 09:08:15 -07:00
Rajat Sen bf23fb4880 readme update and not pinning cuda versin for flax 2025-10-08 15:50:21 +00:00
Rajat Sen cae3104aa6 yml file changes 2025-10-08 00:21:25 +00:00
Rajat Sen 604ad59067 yml file changes 2025-10-08 00:19:55 +00:00
Rajat Sen c172faf5eb yml file changes 2025-10-08 00:14:50 +00:00
Rajat Sen 1c0cd87f2b updating workflows 2025-10-08 00:06:52 +00:00
Rajat Sen 39df49c3a0 uv installation instructions 2025-10-07 23:58:53 +00:00
Rajat Sen 56e7fd67c6 pulled flax changes from master 2025-10-07 21:46:30 +00:00
Rajat Sen 018b2557ab changes to workflow 2025-10-07 21:38:22 +00:00
Yichen Zhou fe70095c6c Merge pull request #323 from google-research/siriuz42-2.0-pr
flax
2025-10-07 14:37:26 -07:00
siriuz42 1436ad859c fix typo 2025-10-07 21:26:57 +00:00
siriuz42 ff2cefaa26 minor changes 2025-10-07 20:41:38 +00:00
siriuz42 25bb4f20e8 (1) Make config hashable. (2) Flax 2025-10-06 05:58:30 +00:00
siriuz42 c159f1bf39 setup pyproject.toml to add optional dependencies 2025-10-03 16:56:53 -07:00
siriuz42 cb3aa09b15 setup pyproject.toml to add optional dependencies 2025-10-03 14:50:31 -07:00
siriuz42 a314cd252c initial flax push 2025-10-03 14:14:36 -07:00
Rajat Sen dc35e580d9 Merge pull request #316 from google-research/rajat-dev
Faster decoding related changes
2025-10-01 15:23:52 -07:00
Rajat Sen a58bde615d minor doc changes 2025-10-01 21:47:42 +00:00
Rajat Sen 4f6bf5adc0 readme update 2025-10-01 18:13:36 +00:00
Rajat Sen 501ddd5614 adding these changes to default config. 2025-10-01 18:10:10 +00:00
Rajat Sen 08918788ad adding these changes to default config. 2025-10-01 18:08:11 +00:00
Rajat Sen c58e6b6bca adding torch attention support and refactoring compile options 2025-10-01 18:05:30 +00:00
Rajat Sen 2b5516d2f1 minor changes to readme 2025-10-01 16:51:59 +00:00
Rajat Sen a1c61b6f5d faster decoding related changes 2025-10-01 16:50:27 +00:00
Yichen Zhou d5a3d5b872 Update README.md 2025-09-26 17:53:11 -07:00
Yichen Zhou 03e3f7a9d8 Merge pull request #312 from google-research/rajat-dev
Download using ModelHubMixin
2025-09-26 17:49:23 -07:00
Rajat Sen 2fded92687 indentation changes 2025-09-26 23:25:06 +00:00
Rajat Sen 1c6e15ff21 update readme 2025-09-26 22:52:18 +00:00
Rajat Sen c42a740917 loading the model using from_pretrained method 2025-09-26 22:47:56 +00:00
Yichen Zhou 2e6db886fa Merge pull request #310 from ram-from-tvl/fix/remove-lingvo
Remove lingvo from v1 pax extra to avoid install failures on Colab
2025-09-25 13:06:00 -07:00
ram-from-tvl 098ad05360 Remove lingvo from v1 pax extra to avoid install failures on Colab 2025-09-25 23:40:10 +05:30
Yichen Zhou b810bbdf9f Merge pull request #299 from google-research/siriuz42-2.0-pr
MVP readme
2025-09-15 09:57:09 -07:00
Yichen Zhou 099cefe8b9 bug fix, docstring, readme 2025-09-13 06:57:00 +00:00
Yichen Zhou b9e31fd02a To comply with Google OSS style guide. 2025-09-13 05:30:07 +00:00
Yichen Zhou 59d2985681 Update timesfm_2p5_torch.py
Update Hugging Face repo id.
2025-09-11 18:17:14 -07:00
Rajat Sen 24dba82664 Merge pull request #297 from google-research/siriuz42-2.0-pr
2.0.0 initial
2025-09-11 17:26:14 -07:00
siriuz42 7d8f3d971d 2.0.0 initial 2025-09-12 00:18:08 +00:00
Rajat Sen d70708d42a Merge pull request #294 from Funto-Adeyemi/add-troubleshooting-section
Add troubleshooting section in README.md and apply yapf formatting
2025-08-29 17:21:59 -07:00