psutil

[cuda]
nvidia-ml-py

[e2e_testing]
accelerate

[mpi]
mpi4py>=3.1.4
mpi_rdv

[rocm]
amdsmi

[rocm-auto]
amdsmi

[testing]
pytest

[torch]
torch
numpy
