chex
einops
numpy
jax
dm-haiku
graphviz
matplotlib
pandas
tenacity
networkx