This NeurIPS 2021 paper seems interesting and relevant. https://openreview.net/pdf?id=xXYjxli-2i
This NeurIPS 2021 paper seems interesting and relevant. https://openreview.net/pdf?id=xXYjxli-2i