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
03e3f7a9d8
Merge pull request #312 from google-research/rajat-dev
...
Download using ModelHubMixin
2025-09-26 17:49:23 -07:00
2fded92687
indentation changes
2025-09-26 23:25:06 +00:00
1c6e15ff21
update readme
2025-09-26 22:52:18 +00:00
c42a740917
loading the model using from_pretrained method
2025-09-26 22:47:56 +00:00
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
098ad05360
Remove lingvo from v1 pax extra to avoid install failures on Colab
2025-09-25 23:40:10 +05:30
b810bbdf9f
Merge pull request #299 from google-research/siriuz42-2.0-pr
...
MVP readme
2025-09-15 09:57:09 -07:00
099cefe8b9
bug fix, docstring, readme
2025-09-13 06:57:00 +00:00
b9e31fd02a
To comply with Google OSS style guide.
2025-09-13 05:30:07 +00:00
Kashif Rasul