PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838 - View it on GitHub