rocm_jax

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

History

Nitin Srinivasan 0159bead97 Move output path to be inside the wheel build command execution loop

This was causing an issue when building multiple wheels in editable mode.

i.e instead of wheels being stored as:
```
# jax-cuda12-pjrt   0.4.36.dev20241125           ./dist/jax-cuda-pjrt
# jax-cuda12-plugin 0.4.36.dev20241125           ./dist/jax-cuda-plugin
# jaxlib            0.4.36.dev20241125           ./dist/jaxlib
```

they were being stored as:
```
# jaxlib            0.4.36.dev20241125           ./dist/jaxlib
# jax-cuda12-pjrt   0.4.36.dev20241125           ./dist/jaxlib/jax-cuda-pjrt
# jax-cuda12-plugin 0.4.36.dev20241125           ./dist/jaxlib/jax-cuda-plugin
```

PiperOrigin-RevId: 708468522

2024-12-20 17:27:51 -08:00

rocm

Merge pull request #25440 from ROCm:jax_pypi-upstream

2024-12-19 09:17:46 -08:00

tools

Add experimental support for building JAX CPU and GPU wheels with GCC.

2024-12-19 09:03:25 -08:00

BUILD.bazel

Use hermetic Python in JAX, see "Managing hermetic Python" in developer.md for details

2024-05-15 18:20:56 -07:00

build.py

Move output path to be inside the wheel build command execution loop