xla icon indicating copy to clipboard operation
xla copied to clipboard

Published 20 hours ago • verified publisher icon openxla

[JAX] Split _src/xla_bridge.py into a separate Bazel target.

Open copybara-service[bot] opened this issue 2 years ago • 0 comments

[JAX] Split _src/xla_bridge.py into a separate Bazel target.

Include _src/distributed.py and _src/clusters/*.py in the same target because they are in a strongly-connected component.

[XLA:Python] Set type of ArrayImpl to Any, since the JAX change now allows pytype to see that some values are ArrayImpls but ArrayImpls are not instances of jax.Array to Pytype.

Fix type of buffer_from_pyval.

copybara-service[bot] avatar Mar 10 '23 14:03 copybara-service[bot]