[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
