numpy
jax
jaxlib
matplotlib
colorama
graphviz
yattag
