Closed copybara-service[bot] closed 3 months ago
Update JAX dependency to >=0.4.26 so that we can access the jax.nn.sparse_plus function.
Update JAX dependency to >=0.4.26 so that we can access the jax.nn.sparse_plus function.