torch>=2.0.0
triton>=2.0.0
transformers>=4.38.0
matplotlib
pytest
