numpy
scipy
pillow>=10.0.0
gemmi
mrcfile
timm
einops
torch>=2.0.0

[train]
wandb
h5py
matplotlib
seaborn
