torch>=2.0.0
timm>=0.6.13
einops>=0.6.0
ruamel.yaml>=0.17.21
