Hyeontaek Lim 36df8619d7 Bump minimum jaxlib version to 0.3.2 and remove transfer guard compatibility code

2022-04-11 15:33:27 +00:00

..

mlir

Integrate LLVM at llvm/llvm-project@eb27da7dec

2022-03-03 08:24:39 -08:00

__init__.py

Bump minimum jaxlib version to 0.3.2 and remove transfer guard compatibility code

2022-04-11 15:33:27 +00:00

xla_bridge.py

Expose AutoSharding's mesh_shape and mesh_ids options to JAX.

2022-04-01 11:47:56 -07:00