jax.export.default_export_platform

jax.export.default_export_platform()

Retrieves the default export platform.

One of: 'tpu', 'cpu', 'cuda', 'rocm'.

Return type:

str