ott.neural.networks.potentials.PotentialTrainState.step

ott.neural.networks.potentials.PotentialTrainState.step#

PotentialTrainState.step: Union[int, jax.Array]#