Thanks for sharing the paper and document. It looks very interesting. Could you maybe also provide Python version of the code?