Code accompanying https://arxiv.org/abs/1802.02219 - View it on GitHub