Do you have a new machine learning model that you want to optimize, and do not want to bother computing the gradients? Autoptim is for you.
As of version 0.3,
pytorch has been replaced with
autograd for automatic differentiation. It makes the interfacing with numpy even simpler.
Autoptim is a small Python package that blends
autograd's automatic differentiation in
The gradients are computed under the hood using automatic differentiation; the user only provides the objective function:
import numpy as np from autoptim import minimize def rosenbrock(x): return (1 - x) ** 2 + 100 * (x - x ** 2) ** 2 x0 = np.zeros(2) x_min, _ = minimize(rosenbrock, x0) print(x_min) [0.99999913 0.99999825]
It comes with the following features:
Natural interfacing with Numpy: The objective function is written in standard Numpy. The input/ output of
autoptim.minimize are Numpy arrays.
Smart input processing:
scipy.optimize.minimize is only meant to deal with one-dimensional arrays as input. In
autoptim, variables can be multi-dimensional arrays or lists of arrays.
Preconditioning: Preconditioning is a simple way to accelerate minimization, by doing a change of variables.
autoptim makes preconditioning straightforward.
This package is meant to be as easy to use as possible. As so, some compromises on the speed of minimization are made.
To install, use
pip install autoptim
Several examples can be found in