# Track A (software prototype) — minimal, runs on CPU or a single GPU.
torch>=2.0
numpy>=1.24
# Optional, only for --dataset mnist:
# torchvision>=0.15