pytorch_lightning
einops
safetensors
transformers
plotly_express
opencv-python
tiktoken
datasets