d2cb4849b6
fix docs string
2026-03-11 17:38:09 +01:00
0134e55a6e
Replace ModelHubMixin with PytorchModelHubMixin
2026-03-11 09:36:14 +01:00
a8bcd5d6d9
use the ModelHubMixin api
2026-03-06 10:53:18 +01:00
8a755c9c75
Merge pull request #341 from kashif/timesfmv1-var
...
[TimesFMv1] fix variance calculation
2026-02-18 16:22:59 -08:00
177ab03e9c
fix v1 jax version
2026-02-18 14:50:56 +01:00
2dcc66fbfe
Merge pull request #360 from google-research/siriuz42-patch-3
...
Update pyproject.toml
2026-01-27 10:53:07 -08:00
4f0bd19a92
Update pyproject.toml
...
support python 3.10
2026-01-27 10:51:19 -08:00
6bd8044275
Merge pull request #345 from google-research/siriuz42-patch-2
...
Update timesfm_2p5_flax.py
2025-12-05 13:48:40 -08:00
f7f6fa5d99
Update timesfm_2p5_flax.py
...
Allow compile to delay the jitting run.
2025-12-05 09:46:19 -08:00
705685c912
fix v1 var calculation
...
1. Masked variance calculation (lines 95-107): Changed from the numerically unstable E[X²] - E[X]² formula to the stable centered formula E[(X-μ)²]
2. Sigma clamping (line 609): Changed from torch.where(sigma < tolerance, 1.0, sigma) to torch.clamp(sigma, min=tolerance)
2025-11-25 09:40:07 +01:00
bf88c5dc88
Merge pull request #335 from google-research/rajat-dev
...
adapt to new huggingface_hub interface
2025-10-31 10:45:25 -07:00
3bb141947e
adapt to new huggingface_hub interface
2025-10-31 17:41:31 +00:00
ddfa6cd9d8
Merge pull request #332 from google-research/rajat-dev
...
manually download config.json
2025-10-30 13:23:28 -07:00
be91ef8ee3
manually download config.json
2025-10-30 20:11:16 +00:00
c43e8f3e66
Update README.md
2025-10-29 17:45:32 -07:00
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
ba2417e11a
update requirement
2025-10-28 22:40:21 +00:00
0d4d5141d7
covariate support
2025-10-28 22:30:24 +00:00
c24db2afd2
covariate support
2025-10-28 22:29:40 +00:00
80f1a1eafe
Merge remote-tracking branch 'origin/master' into siriuz42-2.0-pr
2025-10-28 17:50:55 +00:00
58e01ad82f
Merge pull request #326 from google-research/siriuz42-patch-1
...
Update timesfm_2p5_flax.py
2025-10-08 16:15:28 -07:00
a6609df1d5
Update timesfm_2p5_flax.py
...
bug fix
2025-10-08 16:03:18 -07:00
b6b207a4ed
bug fix
2025-10-08 22:59:07 +00:00
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
9080939cb8
Manually trigger jitting in Flax.
2025-10-08 16:32:41 +00:00
a21b1d5b25
Merge pull request #324 from google-research/rajat-dev
...
Use uv instead of poetry
2025-10-08 09:08:15 -07:00
bf23fb4880
readme update and not pinning cuda versin for flax
2025-10-08 15:50:21 +00:00
cae3104aa6
yml file changes
2025-10-08 00:21:25 +00:00
604ad59067
yml file changes
2025-10-08 00:19:55 +00:00
c172faf5eb
yml file changes
2025-10-08 00:14:50 +00:00
1c0cd87f2b
updating workflows
2025-10-08 00:06:52 +00:00
39df49c3a0
uv installation instructions
2025-10-07 23:58:53 +00:00
56e7fd67c6
pulled flax changes from master
2025-10-07 21:46:30 +00:00
018b2557ab
changes to workflow
2025-10-07 21:38:22 +00:00
fe70095c6c
Merge pull request #323 from google-research/siriuz42-2.0-pr
...
flax
2025-10-07 14:37:26 -07:00
1436ad859c
fix typo
2025-10-07 21:26:57 +00:00
ff2cefaa26
minor changes
2025-10-07 20:41:38 +00:00
25bb4f20e8
(1) Make config hashable. (2) Flax
2025-10-06 05:58:30 +00:00
c159f1bf39
setup pyproject.toml to add optional dependencies
2025-10-03 16:56:53 -07:00
cb3aa09b15
setup pyproject.toml to add optional dependencies
2025-10-03 14:50:31 -07:00
a314cd252c
initial flax push
2025-10-03 14:14:36 -07:00
dc35e580d9
Merge pull request #316 from google-research/rajat-dev
...
Faster decoding related changes
2025-10-01 15:23:52 -07:00
a58bde615d
minor doc changes
2025-10-01 21:47:42 +00:00
4f6bf5adc0
readme update
2025-10-01 18:13:36 +00:00
501ddd5614
adding these changes to default config.
2025-10-01 18:10:10 +00:00
08918788ad
adding these changes to default config.
2025-10-01 18:08:11 +00:00
c58e6b6bca
adding torch attention support and refactoring compile options
2025-10-01 18:05:30 +00:00
2b5516d2f1
minor changes to readme
2025-10-01 16:51:59 +00:00
a1c61b6f5d
faster decoding related changes
2025-10-01 16:50:27 +00:00
d5a3d5b872
Update README.md
2025-09-26 17:53:11 -07:00
Kashif Rasul