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.

Real Cube Toss

Variable Stiffness - MuJoCo simulation

Soft

Medium

Hard

Kubric