NTK-MMD Pytorch code to reproduce the numerical results in X. Cheng and Y. Xie. "Neural Tangent Kernel Maximum Mean Discrepancy". arXiv:2106.03227, 2021 ArXiv link: https://arxiv.org/abs/2106.03227