Skip to content

GitLab

"...git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "1f828f4e5d313f722cd36beef139afe38c452076"
Commit c5c09c6c authored by Phuong Nguyen's avatar Phuong Nguyen Committed by Kshitij Janardan Lakhani
Browse files

[JAX] Update JAX version requirement in pyproject.toml (#2197) 



update jax requirements
Signed-off-by: default avatarPhuong Nguyen <phuonguyen@nvidia.com>
parent 408f0de6
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
pyproject.toml
View file @ c5c09c6c
...@@ -3,7 +3,8 @@ ...@@ -3,7 +3,8 @@
# See LICENSE for license information. # See LICENSE for license information.
[build-system] [build-system]
requires = ["setuptools>=61.0", "cmake>=3.21", "wheel", "pybind11[global]", "ninja", "pip", "torch>=2.1", "jax[cuda12]", "flax>=0.7.1"] requires = ["setuptools>=61.0", "cmake>=3.21", "wheel", "pybind11[global]", "ninja", "pip",
"torch>=2.1", "jax>=0.5.0", "flax>=0.7.1"]
# Use legacy backend to import local packages in setup.py # Use legacy backend to import local packages in setup.py
build-backend = "setuptools.build_meta:__legacy__" build-backend = "setuptools.build_meta:__legacy__"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment