jax

16636f9c - [jax_triton] Only use side stream to do autotuning when doing graph capture

[jax_triton] Only use side stream to do autotuning when doing graph capture When graph capture is not enabled, autotuning and kernel launch should be on the same stream to avoid race condition. PiperOrigin-RevId: 603728867