numpy
omegaconf
einops
transformers
pytorch-lightning
diffusers