Use jax.lax.pure_callback in jax.ScipyMinimize - Githubissues

google / jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

https://jaxopt.github.io

Apache License 2.0

939 stars 66 forks source link

Use jax.lax.pure_callback in jax.ScipyMinimize #372

Open mblondel opened 1 year ago