This paper has been well-received by the reviewers and the author response addressed the reviewers' concerns adequately. Thus, the paper is suitable for acceptance at NeurIPS without significant changes.