1. Department of Statistics and Data Science, Carnegie Mellon University, Pittsburgh, PA 15213, USA
2. Machine Learning Department, Carnegie Mellon University, Pittsburgh, PA 15213, USA
3. McWilliams Center for Cosmology, Carnegie Mellon University, Pittsburgh, PA 15213, USA
4. Department of Statistics and Data Science, Yale University, New Haven, CT 06520, USA
5. Department of Physics, Carnegie Mellon University, Pittsburgh, PA 15213, USA
6. School of Physics, University of Melbourne, Parkville, VIC 3010, Australia