Imported from https://github.com/matthew-hirn/ScatNet-QM-2D