Learning Layer-wise Equivariances Automatically using Gradients Tycho F.A. van der Ouderaa Alexander Immer