jax.export.default_export_platform

jax.export.default_export_platform()

检索默认导出平台。

以下之一：tpu, cpu, cuda, rocm。

返回类型:

str