Before revision:  We demonstrate this framework on a series of challenging 2D and 3D physical systems with different coefficients of restitution and friction  .
After revision:  We demonstrate this framework on a series of challenging 2D and 3D physical systems with different coefficients of restitution and friction . The learned dynamics can be used as a differentiable physics simulator for downstream gradient-based optimization tasks, such as planning and control .