[Bug] Fail to run multi processes mode in Tunix #857
Description
Checklist
- I have searched related issues but cannot get the expected help.
- The bug has not been fixed in the latest version.
- Please note that if the bug-related issue you submitted lacks corresponding environment info and a minimal reproducible demo, it will be challenging for us to reproduce and resolve the issue, reducing the likelihood of receiving feedback.
- If the issue you raised is not a bug but a question, please raise a discussion at https://github.com/sgl-project/sgl-jax/discussions/new/choose Otherwise, it will be closed.
- Please use English, otherwise it will be closed.
Describe the bug
RuntimeError: Unable to initialize backend 'tpu': ABORTED: The TPU is already in use by process with pid 2718466. Not attempting to load libtpu.so in this process. (set JAX_PLATFORMS='' to automatically choose an available backend)
2026-03-03 05:28:44 - WARNING - [sgl_jax.srt.entrypoints.engine] Child process unexpectedly failed with exitcode=256. pid=2720317