jaxtyping
rich>=12
torch

[:python_version < "3.8"]
typing_extensions

[dev]
black[jupyter]==22.3.0
isort==5.10.1
pylint==2.13.4
pytest==7.1.2
pytest-xdist==2.5.0
typeguard>=2.13.3
pyyaml==6.0
build
twine
ninja
