You might like this one; seems to be the basis for the blog post:

- https://arxiv.org/abs/2307.10169