annotated-types
colorama
graphviz
jax
jaxlib
jaxtyping
matplotlib
numpy>=1.23.0
sympy
yattag
