1
Fork 0

fix(hpc): cuda

This commit is contained in:
Tibo De Peuter 2026-04-04 23:35:48 +02:00
parent 7e902aa8b6
commit 6065603fbd
2 changed files with 2 additions and 2 deletions

View file

@ -28,7 +28,7 @@ dependencies = [
[project.optional-dependencies]
cuda = [
"jax[cuda13]==0.9.0.1",
"jax[cuda12]==0.9.0.1",
]
[dependency-groups]