Attempt to fix scipy wrappers to jit or vmap

google / jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

https://jaxopt.github.io

Apache License 2.0

936 stars 66 forks source link

Closed vroulet closed 1 year ago

vroulet commented 1 year ago

Tried to use a pure_callback (see sandbox function) but does not work yet.