Paper Accepted at AISTATS 2022