torch>=2.0.0
transformers<5,>=4.30.0
safetensors>=0.3.0
h5py>=3.8.0
numpy>=1.24.0
scikit-learn>=1.3.0
tqdm>=4.65.0
datasets>=2.12.0
accelerate>=0.20.0
tokenizers>=0.13.0
scipy>=1.10.0
pandas>=2.0.0
python-dotenv>=1.0.1
PyYAML>=6.0

[:python_version >= "3.11"]
numba>=0.60.0
llvmlite>=0.43.0

[nlp]
nltk>=3.8.0

[spacy]
spacy>=3.6.0

[viz]
matplotlib>=3.7.0
seaborn>=0.12.0
umap-learn>=0.5.6

[wandb]
wandb>=0.15.0
