Make jax/tools/pgo nsys converter.py safe to use in local and automated profile-processing workflows.
Make jax/tools/pgo nsys converter.py safe to use in local and automated profile-processing workflows.: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). Make jax/tools/pgo nsys converter.py safe to use in local and automated profile-processing workflows. Today the tool assumes…
The task
Make jax/tools/pgo_nsys_converter.py safe to use in local and automated profile-processing workflows. Today the tool assumes that every conversion starts from an Nsys profile, the advertised --post_process mode has no effect, output is opened before the input has been…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.