jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
921 stars
17 watching
64 forks
Language: Python
last commit: 11 days ago
Linked from 1 awesome list
bi-leveldeep-learningdifferentiable-programmingjaxoptimization
n2cholas/awesome-jax