Permutation-equivariant neural networks applied to dynamics prediction