psutil
numpy
torch >= 2.0.0
transformers >= 4.28.0
mypy
pytest