https://barbatti.org/2018/09/14/nonadiabatic-dynamics-with-machine-learning/