The parameter generator needs to ensure atom ordering symmetries. For example, when predicting bond parameters, if we exchange the order of two input atomic representations, the predicted parameters should be invariant. In the Espaloma model, 41 the relevant equivalent atom permutations are enumerated, which has a computational cost. Here, we leverage this by splitting the input atom embeddings based on their intrinsic structure and applying linear transformations to ensure symmetry: