Interesting... looks like this uses pymagnitude

https://github.com/plasticityai/magnitude