Skip to content
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -176,7 +176,8 @@ pin_pytorch_gpu = [
"torch==2.10.0",
]
pin_jax_cpu = [
"jax==0.5.0;python_version>='3.10'",
"jax>=0.6.2;python_version>='3.10'",
"jax==0.10.0;python_version>='3.11'",
Comment thread
njzjz marked this conversation as resolved.
Outdated
]
pin_jax_gpu = [
"jax[cuda12]==0.5.0;python_version>='3.10'",
Expand Down
Loading