jax.experimental.pallas.tpu.is_tpu_device

Contents

jax.experimental.pallas.tpu.is_tpu_device#

jax.experimental.pallas.tpu.is_tpu_device()[source]#

Returns whether the current device is a TPU.

Return type:: bool