An additional rotation-invariant MatTEN model was similarly created, however, the model was trained on the shift tensor eigenvalues. Internally, this model still does message passing using equivariant embeddings (explained in the following paragraph), however, in this case, the target is set to the (scalar) eigenvalues instead of the full chemical shift tensor.