Implement isotropic box optimization.
Adjust to which inputs/outputs the model supports.

Some features will fail if the model doesn't support calculating a
certain value but the rest will continue to work fine.