Accepted as a poster presentation at NeurIPS 2024. Proceedings: https://proceedings.neurips.cc/paper_files/paper/2024/hash/0f9e0309d8a947ca44463a9b7e8b6a3f-Abstract ...