Skip to content

Set JAX_PLATFORMS to "tpu, cpu" for ray worker#145

Merged
richardsliu merged 1 commit intoAI-Hypercomputer:mainfrom
richardsliu:patch-2
Jul 9, 2024
Merged

Set JAX_PLATFORMS to "tpu, cpu" for ray worker#145
richardsliu merged 1 commit intoAI-Hypercomputer:mainfrom
richardsliu:patch-2

Commits

Commits on Jul 9, 2024