torch>=1.9
lightning
dypy
