Model architecture visualization tool (2D or 3D)

Hello,

Is there any interesting tool to plot the model architecture in 3D or 2D at least ?

Thank you

Tensorboard is one option, have a look at the tutorial for it:
https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html