torch>=2.0.0
# Removed flash-attn for now.
# flash-attn>=2.5.6 --install-option='--no-build-isolation'
pyvene>=0.1.1
transformers>=4.39.3
protobuf>=3.20.0
matplotlib>=3.7.4
ipywidgets>=8.1.1
plotnine>=0.12.4
huggingface-hub>=0.20.3
numpy>=1.26.4
accelerate>=0.29.1
sentencepiece>=0.1.96
evaluate>=0.4.1
datasets>=2.18.0
wandb
scikit-learn
jupyter
# Kaggle notebook setup.
fsspec>=2024.2.0
ydata-profiling>=4.7.0
seaborn==0.12.2
# Colab notebook setup.
gcsfs>=2024.2.0