jax
a4036e4a - [jax.collect_profile] Allow arbitrary options to be passed to XProf

Commit
110 days ago
[jax.collect_profile] Allow arbitrary options to be passed to XProf XProf has a number of underlying profiler options that can be exposed, with more being added. This will limit the number of updates to this script, as well as prevent any incorrect flags from being listed as a result of version differences. PiperOrigin-RevId: 855342653
Author
Parents
Loading