Missing graph visualisation

[edran]

We should write an interface for visualising PyTorch (or any other NN graph).

[ytian81]

Any updates?

[sunshineatnoon]

Any updates? That's a cool function.

[nashory]

Any updates...?

[albanD]

Some experiments have been done to visualize Pytorch graphs in tensorflow using ONNX. The problem is that the graphs usually end up being too big because they print at the Function level.
A way to add higher level scopes for nn Modules for example would be needed to get useful graphs.