Output gradient
###############

.. automodule:: metatrain.utils.output_gradient
    :members:
    :undoc-members:
    :show-inheritance: