ott.math.utils.logsumexp

ott.math.utils.logsumexp = <jax._src.custom_derivatives.custom_jvp object>