torch
transformers
pillow
numpy