Neural-network modules

Activation(irreps_in, acts)	Raw-array compatible Activation.
BatchNorm(irreps[, eps, momentum, affine, ...])	Raw-array compatible BatchNorm.
Dropout(irreps, p)	Raw-array compatible Dropout.
Extract(irreps_in, irreps_outs, instructions)
ExtractIr(irreps_in, ir)
FullyConnectedNet(hs[, act, variance_in, ...])
Gate(irreps_scalars, *args[, ...])	Raw-array compatible Gate.
Identity(irreps_in, irreps_out)	Raw-array compatible Identity.
NormActivation(irreps_in, scalar_nonlinearity)	Raw-array compatible NormActivation.
S2Activation(irreps, act, res[, ...])
SO3Activation(lmax_in, lmax_out, act, ...[, ...])