/NTK-MMD

Neural Tangent Kernel MMD

Primary LanguageJupyter Notebook

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