Switch to a new thread-safe utility for catching warnings.#25626

copybara-service[bot] merged 1 commit intojax-ml:mainjax-ml/jax:mainfrom

hawkinsp:warnings2hawkinsp/jax:warnings2Copy head branch name to clipboard