search

google / jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.
https://jaxopt.github.io
Apache License 2.0
903 stars 62 forks source link

Updated jax.config import #550

Closed copybara-service[bot] closed 8 months ago

copybara-service[bot] commented 8 months ago

Updated jax.config import