ott.neural.networks.potentials.PotentialTrainState.potential_value_fn# PotentialTrainState.potential_value_fn: Callable[[FrozenDict[str, Array], Callable[[Array], Array] | None], Callable[[Array], Array]]#
