xla
e42b9727 - Guard JAX import for TPU tests only

Commit

2 years ago

Guard JAX import for TPU tests only

References

#6747 - [Pallas] Introduce jax_import_guard

Author

alanwaketan

alanwaketan

Committer

alanwaketan

alanwaketan

Parents

Loading