Invertible Gaussian Reparameterization: Revisiting the Gumbel-Softmax. - Andres Potapczynski; Gabriel Loaiza-Ganem; John P. Cunningham (NeurIPS, 2020)

Bias-Free Scalable Gaussian Processes via Randomized Truncations. - Andres Potapczynski; Luhuan Wu; Dan Biderman; Geoff Pleiss; John P. Cunningham (ICML, 2021)

PAC-Bayes Compression Bounds So Tight That They Can Explain Generalization. - Sanae Lotfi; Marc Finzi; Sanyam Kapoor; Andres Potapczynski; Micah Goldblum; Andrew Gordon Wilson (NeurIPS, 2022)

Low-precision arithmetic for fast Gaussian processes. - Wesley J. Maddox; Andres Potapczynski; Andrew Gordon Wilson (UAI, 2022)

On the Normalizing Constant of the Continuous Categorical Distribution. - Elliott Gordon-Rodríguez; Gabriel Loaiza-Ganem; Andres Potapczynski; John P. Cunningham (CoRR, 2022)

A Stable and Scalable Method for Solving Initial Value PDEs with Neural Networks. - Marc Anton Finzi; Andres Potapczynski; Matthew Choptuik; Andrew Gordon Wilson (ICLR, 2023)

Simple and Fast Group Robustness by Automatic Feature Reweighting. - Shikai Qiu; Andres Potapczynski; Pavel Izmailov; Andrew Gordon Wilson (ICML, 2023)

CoLA: Exploiting Compositional Structure for Automatic and Efficient Numerical Linear Algebra. - Andres Potapczynski; Marc Finzi; Geoff Pleiss; Andrew Gordon Wilson (NeurIPS, 2023)

Compute Better Spent: Replacing Dense Layers with Structured Matrices. - Shikai Qiu; Andres Potapczynski; Marc Anton Finzi; Micah Goldblum; Andrew Gordon Wilson (ICML, 2024)

Searching for Efficient Linear Layers over a Continuous Space of Structured Matrices. - Andres Potapczynski; Shikai Qiu; Marc Finzi; Christopher Ferri; Charlie Chen; Micah Goldblum; C. Bayan Bruss; Christopher De Sa; Andrew Gordon Wilson (NeurIPS, 2024)

Searching for Efficient Linear Layers over a Continuous Space of Structured Matrices. - Andres Potapczynski; Shikai Qiu; Marc Finzi; Christopher Ferri; Zixi Chen; Micah Goldblum; C. Bayan Bruss; Christopher De Sa; Andrew Gordon Wilson (CoRR, 2024)

Training Flexible Models of Genetic Variant Effects from Functional Annotations using Accelerated Linear Algebra. - Alan Nawzad Amin; Andres Potapczynski; Andrew Gordon Wilson (CoRR, 2025)