torch>=1.9.0
transformers>=4.20.0
nnsight>=0.2.0
numpy>=1.19.0
inflect>=6.0.0
typing-extensions>=4.0.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=22.0.0
isort>=5.10.0
flake8>=4.0.0
mypy>=0.950
pre-commit>=2.17.0

[docs]
sphinx>=4.5.0
sphinx-rtd-theme>=1.0.0
sphinx-autodoc-typehints>=1.18.0

[examples]
scipy>=1.7.0
matplotlib>=3.4.0
seaborn>=0.11.0
jupyter>=1.0.0
ipywidgets>=7.6.0
