Save the date! Google I/O returns May 18-20 Register now

tfp.experimental.lazybones.utils.plot_graph

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