bson>=0.5.10
auto_mix_prep==0.2.0
datasets==3.0.2
huggingface_hub==0.26.1
numba==0.57.1
numpy==1.24.3
torch==2.1.2
tqdm==4.66.5
transformers==4.32.1

[dev]
pytest>=7.0
twine>=4.0.2
