ott.neural.networks.potentials.PotentialTrainState.apply\_gradients
===================================================================

.. currentmodule:: ott.neural.networks.potentials

.. automethod:: PotentialTrainState.apply_gradients