numpy 
einops 
opt-einsum 
torch 
tqdm 
h5py 
matplotlib 

mypy 
pylint 
pytest 
