jax.export.默认导出平台# jax.export.default_export_platform()[源代码][源代码]# 检索默认的导出平台。 其中之一:tpu、cpu、cuda、rocm。 返回类型: str