torch
torchvision
torchtyping
einops
diffusers
transformers
accelerate
xformers
wandb
fastprogress
matplotlib

[benchmarks]
pygraphviz
graphviz
openai

[dev]
black
isort
flake8
flake8-pyproject
pydocstyle
mypy
pre-commit
pytest
pytest-cov

[docs]
sphinx==5.3.0
sphinx_rtd_theme
sphinx_autodoc_typehints

[notebook]
ipython

[sodaracer]
swig>=4.1.0
box2d-py==2.3.8
pygame

[triton]
tritonclient[all]
