setup
compact
pjit
Gated linear unit activation function.
x – input array
axis – the axis along which the split should be computed (default: -1)