qm_egnn.zip