I hope this post finds you in perfect health condition. I am trying to implement this paper https://arxiv.org/abs/2003.07833 and the code is publicly available. I want to ap