torch>=2.0
accelerate==0.29.2
simple-parsing
