mirror of
https://github.com/ROCm/jax.git
synced 2025-04-24 19:46:07 +00:00

The jaxlib wheel build currently uses `python setup.py bdist_wheel` to construct the wheel. Change it to use `python -m build -w` instead. To avoid Python getting confused between the directory named `build` in the bazel tree and the Python `build` module, move `build_wheel.py` into `jaxlib/tools`. PiperOrigin-RevId: 548133811