fix(hpc): cuda
This commit is contained in:
parent
7e902aa8b6
commit
6065603fbd
2 changed files with 2 additions and 2 deletions
|
|
@ -28,7 +28,7 @@ dependencies = [
|
|||
|
||||
[project.optional-dependencies]
|
||||
cuda = [
|
||||
"jax[cuda13]==0.9.0.1",
|
||||
"jax[cuda12]==0.9.0.1",
|
||||
]
|
||||
|
||||
[dependency-groups]
|
||||
|
|
|
|||
Reference in a new issue