Implements the ideas presented in https://arxiv.org/pdf/2004.11362v1.pdf by Khosla et al. - View it on GitHub