Replace jax.lib.xla_client._xla with jax.lib.xla_extension#4838
Open
copybara-service[bot] wants to merge 1 commit intomainfrom cl/659607140
+2-1
jax.lib.xla_client._xla with jax.lib.xla_extension#4838
