Equivariant non-linear maps for neural networks on homogeneous spaces