where e x (r) is the exchange energy density coming from the same gauge as e MP2 c (r) (we use the same Python code to implement both e MP2 c (r) and e x (r) on the same footing). Finally, w c (r) are the ML2 weights obtained from the NN (see Fig. ) for the illustration and Sec. VI for NN architecture details).