search

google / jaxopt

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

Use optax losses as backend for jaxopt losses. #586

Open copybara-service[bot] opened 8 months ago

copybara-service[bot] commented 8 months ago

Use optax losses as backend for jaxopt losses.