Relax JAX version pins and add jax[cuda12] to and-cuda extra - #1813
Open
copybara-service[bot] wants to merge 1 commit into
Open
Relax JAX version pins and add jax[cuda12] to and-cuda extra#1813copybara-service[bot] wants to merge 1 commit into
copybara-service[bot] wants to merge 1 commit into