Skip to content
Snippets Groups Projects
Commit d55b56ec authored by Nitin Srinivasan's avatar Nitin Srinivasan
Browse files

Update jax and jaxlib for M1 compatibility

parent 5ede2e3c
Branches
No related tags found
No related merge requests found
......@@ -39,8 +39,8 @@ grpcio ~= 1.42.0
portpicker ~= 1.4.0
scipy ~= 1.7.3
# Required for TFLite import from JAX tests
jax ~= 0.2.26
jaxlib ~= 0.1.75
jax ~= 0.3.14
jaxlib ~= 0.3.14
# Needs to be addressed. Unblocked 2.4 branchcut cl/338377048
PyYAML ~= 5.4.1
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment