torch>=1.12
numpy>=1.22
h5py>=3.7
numba>=0.55
pytorch-lightning
tqdm
sympy
pyyaml
opt_einsum
scikit-learn
