MixMax: Distributional Robustness in Function Space via Optimal Data Mixtures - Thudi, Anvith; Maddison, Chris J. (2024)