The performance and capability of SpaiNN is demonstrated using simulations of the methylenimmonium cation and a series of alkenes. The results show improved accuracy and comparable computational efficiency to previous methods (SchNarc), particularly in fitting non-adiabatic couplings, energies, and forces of multiple electronic excited singlet states. Overall, we show that the usage of the equivariant representation, improves the data efficiency in training and inference as well as the generalization of the respective models. We believe that SpaiNN is promising for the study of various photophysical and photochemical phenomena and the comprehensive documentation and tutorials available with this method facilitate the development of new interatomic potentials for molecules in their excited states.