Commit Graph

253 Commits

Author SHA1 Message Date
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
Rajat Sen ba034ae71c Merge pull request #283 from misha-chertushkin/master
Add safetensors loading from local environment
2025-07-07 15:12:13 -07:00
Rajat Sen 95bdf54d50 Update main.yml 2025-07-06 15:06:46 -07:00
Rajat Sen b0b9329064 Update pyproject.toml 2025-07-06 15:06:23 -07:00
Rajat Sen 756217d03b Merge pull request #285 from AzulGarza/feat/allow-more-freqs
feat: allow more frequencies
2025-07-06 15:05:16 -07:00
azul fffb37852b feat: allow more frequencies 2025-07-04 15:55:11 -07:00
misha-chertushkin 6cf4f23675 Remove yfinance 2025-06-25 16:37:08 +00:00
misha-chertushkin e28239cb2f Added flag for safetensors loading 2025-06-25 16:33:45 +00:00
Rajat Sen 5950ef0653 Merge pull request #270 from google-research/dependabot/pip/torch-2.3.1
Bump torch from 2.1.0 to 2.3.1
2025-04-18 10:26:02 -07:00
dependabot[bot] 28f6e0244e Bump torch from 2.1.0 to 2.3.1
Bumps [torch](https://github.com/pytorch/pytorch) from 2.1.0 to 2.3.1.
- [Release notes](https://github.com/pytorch/pytorch/releases)
- [Changelog](https://github.com/pytorch/pytorch/blob/main/RELEASE.md)
- [Commits](https://github.com/pytorch/pytorch/compare/v2.1.0...v2.3.1)

---
updated-dependencies:
- dependency-name: torch
  dependency-version: 2.3.1
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
2025-04-18 17:24:07 +00:00
Rajat Sen 8b7093a55a Merge pull request #265 from Killer3048/feature/lazy-import-xreg
Implement Lazy import for `xreg` dependencies in `TimesFM`
2025-04-10 14:01:07 -07:00
Killer3048 6da174d703 add information about install pax/paxlib for covariate support in section Covariates Support 2025-04-10 10:47:44 +07:00
Killer3048 b1c6f4d925 update readme.md 2025-04-10 09:52:58 +07:00
Killer3048 6678ba6cee update pyproject (exclude jax/pax) and update README 2025-04-10 09:47:07 +07:00
Killer3048 373a5293df change line for smoother_arr 2025-04-09 13:21:04 +07:00
Killer3048 9a520bb940 Implement lazy import for xreg dependencies in forecast_with_covariates to avoid unnecessary JAX installation 2025-04-09 12:55:04 +07:00
Rajat Sen cec7fb4d43 Update main.yml with version 1.2.9 2025-03-06 16:14:57 -08:00
Rajat Sen 898fbcb00d Merge pull request #249 from misha-chertushkin/bump-version-129
Version bump
2025-03-06 16:07:49 -08:00
misha-chertushkin afe814e31f Version bump 2025-03-07 00:04:41 +00:00
Rajat Sen 389980aae3 Merge pull request #247 from misha-chertushkin/fix-finetuning-package
Fix the missing finetuning package
2025-03-05 13:19:34 -08:00
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