functorch
functorch copied to clipboard
add optimization subpackage in `functorch`
Jax has its optimization subpackage in jax.scipy.optimize
but functorch
does not, this repo may provide some help.
You're right alibool. Reuben Feinman's library Pytorch-minimize is awesome. Functorch should have something equivalent to jsoptimize