The general architecture of SpaiNN is illustrated in Fig. . SpaiNN offers several improvements over its predecessor, SchNarc , most notably by supporting a rotationally equiv-ariant representation of molecules and materials using the PaiNN representation in addition to the translationally and rotationally invariant SchNet representation. Furthermore, the code is entirely modular, similar to SchNetPack2.0 , allowing for straightforward addition and modification of modules for predicting excited-state properties.