[Feature]: AutoDeploy: graph visualization tooling

[lucaslie]

🚀 The feature, motivation and pitch

AutoDeploy used to have a working integration with the google model explorer tool for graph visualization, see here: https://github.com/NVIDIA/TensorRT-LLM/blob/main/tensorrt_llm/_torch/auto_deploy/transform/library/visualization.py

The integration hasn't been maintained for a long time and is currently broken.

The task is to revive the integration and update it with the latest visualization feature from model explorer

Alternatives

No response

Additional context

No response

Before submitting a new issue...

• Make sure you already searched for relevant issues, and checked the documentation and examples for answers to frequently asked questions.