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.