Google I/O returns May 18-20! Reserve space and build your schedule Register now

tfp.experimental.lazybones.utils.plot_graph

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