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