Extended Data Fig. 2: A computation graph of the physics simulation. | Nature Machine Intelligence