# ===== Core =====
numpy
scipy

# ===== PyTorch (CUDA 12.4) =====
# Install via: pip install torch torchvision --index-url https://download.pytorch.org/whl/cu124
torch>=2.4
torchvision

# ===== Data loading =====
webdataset
opencv-python

# ===== Training / logging =====
tensorboard
tqdm
matplotlib