numpy
pandas
pyyaml

[judge]
torch
transformers>=4.46

[test]
pytest
torch
transformers>=4.46
scikit-learn

[train]
torch
transformers>=4.46
peft
datasets
accelerate
scikit-learn
