SchNetPack 2.0: A neural network toolbox for atomistic machine learning