torch~=2.1.2
h5py~=3.10.0
pandas~=2.2.0
numpy~=1.26.2
matplotlib~=3.8.2
wandb~=0.16.2
torchmetrics~=1.3.0.post0