Graph Neural Networks for Learning Equivariant Representations of Neural Networks