Enable visualizing the compute graph

[tostenzel]

See

• https://github.com/tinygrad/tinygrad/blob/master/tinygrad/graph.py
• https://github.com/karpathy/micrograd/blob/master/trace_graph.ipynb