Migrate the trace viewer as part of `mlfoundations` + improvements.
Migrate the trace viewer as part of
mlfoundations+ improvements.