termcolor
einops
h5py
torch-pruning == 1.2.2