accelerate
numpy
torchvision
torch
tqdm
einops

[dev]
build
twine

[examples]
einops
diffusers
transformers
datasets
jupyter
matplotlib
torch_ema

[test]
pytest
diffusers==0.27.2
