add optimization subpackage in `functorch`

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.

https://pytorch.org/functorch/

BSD 3-Clause "New" or "Revised" License

1.38k stars 102 forks source link

Open alibool opened 1 year ago

alibool commented 1 year ago

Jax has its optimization subpackage in jax.scipy.optimize but functorch does not, this repo may provide some help.

richinex commented 1 year ago

You're right alibool. Reuben Feinman's library Pytorch-minimize is awesome. Functorch should have something equivalent to jsoptimize