mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move protobuf deps to be optional.

Browse Source 
PiperOrigin-RevId: 466123822
This commit is contained in:
Parker Schuh 2022-08-08 12:57:16 -07:00 committed by jax authors
parent 53f16b7709
commit 022fedde98
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@ -70,7 +70,6 @@ setup(
'scipy>=1.5',
'typing_extensions',
'etils[epath]',
'protobuf>=3.13,<4',
],
extras_require={
# Minimum jaxlib version; used in testing.
@ -90,6 +89,9 @@ setup(
# Required by cloud_tpu_init.py
'requests'],
# $ pip install jax[australis]
'australis': ['protobuf>=3.13,<4'],
# CUDA installations require adding jax releases URL; e.g.
# Cuda installation defaulting to a CUDA and Cudnn version defined above.
# $ pip install jax[cuda] -f https://storage.googleapis.com/jax-releases/jax_releases.html