is_tpu_installed#

composer.utils.is_tpu_installed()[source]#

Determines whether the module needed for training on TPUsโ€”torch_xlaโ€”is installed.

Returns

bool โ€“ Whether torch_xla is installed.