Skip to content

Add support for traces generated by Jax 8.0 #456

@devalshahamd

Description

@devalshahamd

Currently tensorboard-plugin-profile==2.19.0 is used for processing Jax traces (xplane.pb) file, which is not compatible with traces generated by jax 8.0. We need to upgrade to xprof=2.21.3 for these traces, which might require some changes in the TraceLens.

Metadata

Metadata

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions