jax.nn.glu

jax.nn.glu(x, axis=-1)[source]

Gated linear unit activation function.