numpy
more_itertools
array-api-compat

[dev]
pytest
nbval
ml_dtypes
jaxlib
jax
torch
array-api-strict
torchao==0.10
airium
pandas
matplotlib
pre-commit
black
mypy
black[jupyter]
isort
sphinx==7.1.2
sphinx-rtd-theme==1.3.0rc1
sphinx_paramlinks
myst_nb
