Thanks! This came out after this blog but https://arxiv.org/abs/2402.01817 may be of interest too