tfp.experimental.lazybones.utils.plot_graph

Plots leaves and ancestors. (See help(nx.draw_networkx) for kwargs).