Peter Hawkins dfb32fc047 [JAX] Do not build the GPU or TPU extensions if not enabled in the JAX build. ...

There's no point building extensions that cannot be used, and they sometimes cause problems (e.g., TPU support on Windows builds).

Fixes https://github.com/google/jax/issues/10687

PiperOrigin-RevId: 449287997

2022-05-17 12:32:03 -07:00

..

mlir

Adds a wrapper to sparse tensor dialect, as part of an

2022-04-21 11:48:44 -07:00

__init__.py

Fix jax 0.3.11 GPU breakge when used with jaxlib 0.3.10.

2022-05-16 00:24:04 +00:00

xla_bridge.py

[JAX] Do not build the GPU or TPU extensions if not enabled in the JAX build.

2022-05-17 12:32:03 -07:00