mlx.nn.relu2

class relu2(x)

Applies the ReLU² activation function.

Applies \(\max(0, x)^2\) element wise.