Learning Invariant Representations with a Nonparametric Nadaraya-Watson Head Alan Q. Wang Cornell University Minh Nguyen Cornell University Mert R. Sabuncu Cornell University