annotated-types
colorama
graphviz
jaxlib~=0.6.0
jaxtyping
jax~=0.6.0
matplotlib
numpy>=2.1.0
sympy
yattag
