<p dir="auto">Hi, Currently I found the step to JAX on multi-host TPU Cloud, but I do not find the step to Run JAX on multi-host GPU platforms. Does anyone can help?</p>
<p dir="auto">I don't have a node with 8 gpus. I have two nodes each with 4 gpus. So is it possible to train a model on multiple nodes?</p>