Interaction Networks for Learning about Objects, Relations and Physics. Pytorch
github.com