This's an implementation of deepmind Visual Interaction Networks paper using pytorch - View it on GitHub