Pytorch Implementation of Learning Mesh-based Simulation With Graph Networks

This is the unofficial implementation of the approach described in the paper:

Tobias Pfaff, Meire Fortunato, Alvaro Sanchez-Gonzalez and Peter W. Battaglia Learning Mesh-based Simulation With Graph Networks. In ICLR, 2021.

Work in progress.

Current progress: able for long-term rollout with consideration of (self-)collision.