Architecture Diagrams

[alameer]

Please can anyone recommend an application for creating architecture diagrams specific for PyTorch models? Many Thanks in advance