Semi-equivariant GNN architectures for jet tagging