torch>=2.11.0
torchvision>=0.26.0
accelerate>=1.14.0
torchmetrics>=1.8.2
pydantic>=2.12.4
tqdm>=4.67.1
jinja2>=3.1.6
click>=8.3.1
PyYAML>=6.0.3

[mlflow]
mlflow>=3.15.1

[wandb]
wandb>=0.28.0
