ott.neural.networks.potentials.PotentialTrainState.params

PotentialTrainState.params: core.FrozenDict[str, Any]