Comment by kashifr

Check out my pytorch reproduction of the paper here for those interested: https://github.com/NVIDIA/physicsnemo/pull/1660