Take a look at this one where we do it correctly and rigorously https://
arxiv.org/abs/2302.07400

→ View original post on X — @animaanandkumar