Skip to content

GitLab

Unverified Commit afd15a16 authored by Phuong Nguyen's avatar Phuong Nguyen Committed by GitHub
Browse files

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



update jax requirements
Signed-off-by: default avatarPhuong Nguyen <phuonguyen@nvidia.com>
parent 3f875fb5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
pyproject.toml
View file @ afd15a16
...@@ -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