Convert Pytorch Model to Publishable Figure

How do most people create figures for their models? I can visualize my model structure in Tensorboard and would now love to export it into a clean SVG or pdf.

Few screenshots of the encoder and decoder taken from Tensorboard.

encoder:

decoder:

I can’t speak for most people but I have an ad-hoc tool for creating dot that I use with JITed models.
I found that in the end, I always want to fiddle with the detail level so I never made it a generic thing. On the other hand making it [interactive (double click to expand and reduce) seems possible but quite a bit of work.

Best regards

Thomas