torch>=2.3.0
torchtune==0.2.1
#pytorch-ignite
torchmetrics
#torchao
scipy
einops
hydra-core
pytorch_lightning
lightning
lightning-bolts
timm
pyrootutils
tensorboard
matplotlib
pfio