Here is our version we did about a year ago: https://arxiv.org/abs/2401.10910