Neural relational inference for interacting systems - pytorch - View it on GitHub