torch>=1.8.0
scipy
pynvml
hydra-joblib-launcher
torchvision>=0.9.0
numpy>=1.19.0
matplotlib>=3.3.0
wandb>=0.12.0
tqdm>=4.50.0
nbformat>=5.1.0  # For notebook processing
hydra-core>=1.3.0
omegaconf>=2.3.0
hydra-colorlog>=1.2.0  # Optional for colored logging
pandas
seaborn