torch
transformers
gradio
Pillow
numpy
jax
flax