taming-transformers
omegaconf
einops