psutil
amdsmi
nvidia-ml-py

[e2e_testing]
accelerate

[mpi]
mpi4py>=3.1.4
mpi_rdv

[testing]
pytest

[torch]
torch
numpy
