numpy
torch
pandas
nltk>=3.8.2
spacy
scikit-learn
matplotlib
tqdm
ipywidgets
transformers
datasets
huggingface-hub

[optional]
flash-attn
accelerate

[test]
pytest
