PyYAML
jinja2
tiktoken
pydantic>=2
Pillow

[dev]
pytest
black
isort
mypy
ruff

[torch]
torch
fairscale
fire
blobfile
