Argh, this is the paper I meant to link: https://arxiv.org/pdf/2008.06537.pdf

(Updated the above comment as well)