transformers
torch
torchvision
torchaudio

black
pylint
sentencepiece
accelerate
datasets
bitsandbytes
# flash-attn #--no-build-isolation # you may need to manually build this
matplotlib
numpy
pandas
ray # not sure if this is needed..
scipy

# maybe
nltk