torch>=2.5.1
torchvision>=0.20.1
accelerate>=1.2.0
torchmetrics>=1.6.0
pydantic>=2.10.3
tqdm>=4.67.1
wandb>=0.18.7
jinja2>=3.1.4
click>=8.1.7
