libcst>=1.2.0
click>=8.1.7
rich>=13.7.1

[shapes]
jaxtyping

[tests]
pytest
mypy
numpy
ml_dtypes
jaxtyping
torch
