Here, r i is the position of atom i. P i is the model prediction which can be interpreted as an atomic charge. To enforce a total system charge of zero, a charge neutrality constraint is enforced. When PiNet2 is used as the backend, 1 P i is still a scalar but contains information from the higher-order geometrical features via Eq. 4.