[feature] MegaScope Tensor Tracer#3606

Open

superay-a wants to merge 10 commits intoNVIDIA:devfrom

MegatronAPPteam:ztr/megascope_tensor_tracer

Commits on Feb 21, 2026

feat: add tensor tracer functionality and websocket server for training visualization

superay-a
authored and
Cs1799205202
committed
refactor: update compressor configuration handling in training modules

superay-a
authored and
Cs1799205202
committed
feat: add support for pipeline parallel worker connections in tensor tracer websocket server

superay-a
authored and
Cs1799205202
committed
feat: enable tensor tracing only in forward step

superay-a
authored and
Cs1799205202
committed
chore: make tensor tracer optional dependency
Cs1799205202
committed
style: tighten tensor tracer implementation
Cs1799205202
committed
test: add tensor tracer unit coverage
Cs1799205202
committed
fix: make empty compressor gather-safe
Cs1799205202
committed

Commits on Feb 25, 2026

feat: add InputTokens flag and corresponding hook for tensor tracing, and remove max_size limit for websocket connections in training_wsserver.py
superay-a
committed

Commits on Feb 26, 2026

doc: add Tensor Tracer documentation and update related components for improved functionality
superay-a
committed