torch>=2.0.0
lightning>=2.0.0
numpy
einops
npy-patcher
