freedak/timesfm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #45 from google-research/rajat_dev

Browse Source 
Pining jax version.
This commit is contained in:
Rajat Sen
2024-05-21 13:36:15 -07:00
committed by GitHub
parent 469b7071dc 40f2559920
commit e67b824e76
5 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
environment.yml
+1 -2
View File
@@ -13,6 +13,5 @@ dependencies:
- utilsforecast - utilsforecast
- praxis - praxis
- paxml - paxml
- -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html - jax[cuda12]==0.4.26
- "jax[cuda12_pip]"
- einshape - einshape
environment_cpu.yml
+1 -1
View File
@@ -13,5 +13,5 @@ dependencies:
- utilsforecast - utilsforecast
- praxis - praxis
- paxml - paxml
- jax[cpu] - jax[cpu]==0.4.26
- einshape - einshape
experiments/environment.yml
+1 -2
View File
@@ -20,6 +20,5 @@ dependencies:
- git+https://github.com/amazon-science/chronos-forecasting.git - git+https://github.com/amazon-science/chronos-forecasting.git
- praxis - praxis
- paxml - paxml
- -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html - jax[cuda12]==0.4.26
- "jax[cuda12_pip]"
- einshape - einshape
experiments/environment_cpu.yml
+1 -1
View File
@@ -20,5 +20,5 @@ dependencies:
- git+https://github.com/amazon-science/chronos-forecasting.git - git+https://github.com/amazon-science/chronos-forecasting.git
- praxis - praxis
- paxml - paxml
- jax[cpu] - jax[cpu]==0.4.26
- einshape - einshape
pyproject.toml
+1 -1
View File
@@ -8,7 +8,7 @@ dependencies = [
"einshape>=1.0.0", "einshape>=1.0.0",
"paxml>=1.4.0", "paxml>=1.4.0",
"praxis>=1.4.0", "praxis>=1.4.0",
"jax>=0.4.26", "jax==0.4.26",
"numpy>=1.26.4", "numpy>=1.26.4",
"pandas>=2.1.4", "pandas>=2.1.4",
] ]