remove jarrett and _make_graphviz, bitrot
remove jarrett and _make_graphviz, bitrot

might want to revive jarrett later!

google/jax,google/jax,tensorflow/probability,tensorflow/probability,google/jax,google/jax