ott.neural.layers.PositiveDense.rectifier_fn