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