jax.experimental.pallas.tpu.get_tpu_info

jax.experimental.pallas.tpu.get_tpu_info#

jax.experimental.pallas.tpu.get_tpu_info = <jaxlib.weakref_lru_cache.StrongLRUCache object>[source]#