torch
transformers
diffusers
accelerate
numpy