torch>=2.0.0
torch-directml>=0.2.5
