h5py==3.7.0
matplotlib==3.5.2
mpi4py==3.1.3
numpy==1.22.3
ruamel.yaml==0.17.21
scipy==1.6.3
torch==1.12.0
torchvision==0.13.0
wandb==0.15.3
