Paper Accepted at NeurIPS 2023