https://barbatti.org/2022/09/28/excited-state-dynamics-with-machine-learning/