Implementation of RigidFormer, Learning Rigid Dynamics using Transformers - View it on GitHub