* More precise imports

* Fix gradient of `minimize` and `root` wrt higher-dimensional arguments

Also:
* Allow several inputs to be optimized
* Handle disconnected and undefined gradients

---------
Co-authored-by: ricardoV94 <ricardo.vieira1994@gmail.com>