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

Jit update method and remove jit="auto". #512

Closed mblondel closed 1 year ago

mblondel commented 1 year ago

This pull-request adds support for automatically jit compiling the update method. It also removes jit="auto", which was made unnecessary thanks to the jittable verbose support, added in #503.