Hardware accelerated, batchable and differentiable optimizers in JAX.
Here you can share your experience with the project you are suggesting or its comparison with jaxopt. Optional.
A valid email to send you a verification link when necessary or log in.
jaxopt