Comment by bravura

Go to arXiV and download the latex source. Plug it into ChatGPT and ask it to implement in pytorch.

I've really enjoyed this approach.