interactionnetworkpytorch

Pytorch Implementation of Interaction Networks for Learning about Objects, Relations and Physics.