Open Source · ML Infrastructure

Standalone TensorBoard

Decoupling TensorBoard from TensorFlow so any deep-learning framework can inspect training runs without lock-in.

TensorBoard visualization toolkit — scalars, graphs, and experiment tracking UI
TensorBoard: the visualization layer that standalone TensorBoard made framework-agnostic.

Problem

In the mid-2010s, TensorBoard was the de facto tool for watching deep-learning experiments — but it was tightly coupled to TensorFlow. Teams on MXNet, PyTorch, and other stacks either rebuilt visualization from scratch or abandoned inspectability. Explainability infrastructure should not force a framework choice.

Approach

I created standalone TensorBoard (dmlc/tensorboard) as primary author (~6K lines, 49 commits). The core idea: extract the event-file protocol and UI into a framework-agnostic layer that any trainer can write into.

AWS mxboard GitHub repository — logging MXNet data for visualization in TensorBoard
AWS mxboard — official MXNet logging library built on the standalone TensorBoard architecture.

That architecture became the basis for AWS’s official mxboard, was referenced by PyTorch’s widely used tensorboardX, and was publicly acknowledged by Google’s TensorBoard engineering leadership.

Rajat Monga of Google TensorFlow thanking @zihaolucky for standalone TensorBoard community work on GitHub issue #50
Rajat Monga (then leading TensorFlow / TensorBoard at Google) publicly thanking @zihaolucky on dmlc/tensorboard#50.

Impact

← Back to home · All projects