How are you generating the architecture graph from model configs, and do you plan to surface tensor shapes or layer-level parameter counts as well?
[flagged]