torch>=2.0.0
rich
tensorboard
matplotlib
seaborn
numpy
scikit-learn
einops
tokenizers
transformers
safetensors
accelerate
lpips
