ott.neural.networks.layers.posdef.PosDefPotentials.rectifier_fn