torch
torchvision
torchaudio
numpy>=1.25.0
pillow
safetensors>=0.4.2
scipy
einops
transformers>=4.50.3
tokenizers>=0.13.3
sentencepiece
typing_extensions
tqdm
psutil
filelock
requests
torchsde
pyyaml
blake3
simpleeval>=1.0.0

[dev]
pytest>=7.0
pytest-asyncio
