Learning Invariant Representations with a Nonparametric Nadaraya-Watson Head | Read Paper on Bytez