Graph network based simulators can learn discontinuous, rigid contact dynamics

Rollout examples for the different datasets. Model predictions are shown in blue, ground truth predictions in green.