Code accompanying paper, Forward Prediction for Physical Reasoning - View it on GitHub