ICML paper about that: https://proceedings.mlr.press/v235/tkachenko24a.html