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.
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.