torch
ml-collections
diffusers
transformers