ed2 for pytorch This is a pytorch implementation of the electrondensity2 Authors: Zidi Wang, Tao Zhang