jax.export.default_export_platform

jax.export.default_export_platform()

Retrieves the default export platform.

One of: tpu, cpu, cuda, rocm.

Return type:

str