einops
torchtyping
lightning
transformers
datasets
pydantic
pytest-xdist
imageio[ffmpeg]
ujson
torchattacks
wandb
paramiko
matplotlib
scikit-learn