ott.neural.networks.potentials.PotentialTrainState.replace# PotentialTrainState.replace(**updates)# Returns a new object replacing the specified fields with new values.
