mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 03:46:06 +00:00

Description: - Added `py::mod_gil_not_used()` to `PYBIND11_MODULE` for `_triton_ext` and `_tpu_ext`. Refs: - https://py-free-threading.github.io/porting/#__tabbed_1_2 Context: - https://github.com/google/jax/issues/23073