Code used in https://arxiv.org/abs/2010.03080 - View it on GitHub