accelerate
idr_torch>=2.0.1
packaging
torch
