numpy
torch
diffusers
transformers