# Requirements for tests (see "requirements-direct.txt" also for direct dependencies)
pytest
nbval
ml_dtypes
jaxlib
jax
torch
array-api-strict
torchao==0.10
airium
pandas
matplotlib

# Requirements for development
pre-commit
black
mypy
black[jupyter]
isort

# Requirements for docs
sphinx==7.1.2
sphinx-rtd-theme==1.3.0rc1
sphinx_paramlinks
myst_nb
