torch
torchvision
numpy
transformers
peft
protobuf
sentencepiece
pytest
matplotlib
scikit-learn
scipy
spacy
einops
jaxtyping
diffusers
opencv-python
git+https://github.com/facebookresearch/pytorch3d.git