matplotlib
numpy
packaging
torch>=2.3
tqdm

[dev]
captum[optional,test]
sphinx<8.2.0
sphinx-autodoc-typehints
sphinxcontrib-katex

[optional]
openai
scikit-learn
annoy

[test]
pytest
pytest-cov
parameterized
black==25.11.0
flake8
pyre-check-nightly==0.0.101750936314
usort==1.1.0
ufmt

[tutorials]
torchtext
torchvision
