numpy
pandas
pyyaml

[rerank]
torch
transformers>=4.46
peft
trl>=0.12
datasets
accelerate

[retrieve]
torch
transformers>=4.46
peft
accelerate

[test]
pytest
torch
polars
pyarrow

[train]
torch
transformers>=4.46
peft
trl>=0.12
datasets
accelerate
