jaxopt

Python ★ 1.0k updated 16d ago

Hardware accelerated, batchable and differentiable optimizers in JAX.

No plain-English explanation yet — one is being written right now. Check back in a minute.