A PyTorch annotated replication of the paper: https://arxiv.org/abs/2006.10637 - View it on GitHub