[JAX] Remove uses of jax.lib.xla_client. - Githubissues

google-research / scenic

Scenic: A Jax Library for Computer Vision Research and Beyond

Apache License 2.0

3.34k stars 441 forks source link

[JAX] Remove uses of jax.lib.xla_client. #1119

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 1 month ago

[JAX] Remove uses of jax.lib.xla_client.