Repository for NeurIPS 2019 paper - View it on GitHub