A multitask constrained D-MPNN for atom and bond descriptor prediction has previously been developed by Guan et al. Their implementation allows a D-MPNN to be trained on multiple atom and bond properties simultaneously. An attention-based constraining method is used to correct for the discrepancy between the sum of a predicted atom property for all atoms (or predicted bond property for all bonds, where applicable) in a molecule and its expected net value at the molecular level, such as the relationship between atomic partial charges and the net charge of a molecule. However, Guan et al.'s code implementation supports only neutral species, and constraints can only be applied to atom properties but not to bond properties. Moreover, the total loss across different targets is not scaled automatically;