Skip to content

Commit 9847afb

Browse files
njzjzCopilot
andauthored
CI(deps): Update jax (cpu) version to 0.10.0 in pyproject.toml (deepmodeling#5442)
<!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit * **Chores** * Updated JAX dependency constraints to provide targeted versions for Python 3.10 and Python 3.11, ensuring compatible and optimized runtime behavior for each Python release. <!-- review_stack_entry_start --> [![Review Change Stack](https://storage.googleapis.com/coderabbit_public_assets/review-stack-in-coderabbit-ui.svg)](https://app.coderabbit.ai/change-stack/deepmodeling/deepmd-kit/pull/5442) <!-- review_stack_entry_end --> <!-- end of auto-generated comment: release notes by coderabbit.ai --> --------- Signed-off-by: Jinzhe Zeng <jinzhe.zeng@ustc.edu.cn> Signed-off-by: Jinzhe Zeng <njzjz@qq.com> Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
1 parent f5c6394 commit 9847afb

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,8 @@ pin_pytorch_gpu = [
176176
"torch==2.10.0",
177177
]
178178
pin_jax_cpu = [
179-
"jax==0.5.0;python_version>='3.10'",
179+
"jax==0.6.2; python_version=='3.10'",
180+
"jax==0.10.0; python_version>='3.11'",
180181
]
181182
pin_jax_gpu = [
182183
"jax[cuda12]==0.5.0;python_version>='3.10'",

0 commit comments

Comments
 (0)