I need to make a plot/graphic of my neural network. I know there’s stuff out there for plotting graphs but I can’t seem to find anything that has the typical structure with the vertices lined up and connected by edges to subsequent layers. Like so…

I’ve found a lot of generic things to make this work but I need to customize edges. Is there any method of visualizing neural networks others here are using? At this point I can only think of learning some visualization software like photoshop or exporting my Flux model to Tensorflow and using tensorboards.