skypilot icon indicating copy to clipboard operation
skypilot copied to clipboard

Published 20 hours ago • verified publisher icon skypilot-org

Add support for TPU VMs

Open gmittal opened this issue 2 years ago • 1 comments

Currently Sky only supports TPU nodes, but an investigation by @infwinston and myself with our JAX experiments showed that Google is pushing all support to TPU VMs (see: https://cloud.google.com/blog/products/compute/introducing-cloud-tpu-vms) for JAX/newer frameworks (even Colab is backed by TPU VMs now). It is important that we support these types of machines as JAX is not compatible with the older TPU nodes.

iirc there was a discussion about a team at Berkeley needing TPU nodes, so we should find a way to support both.

gmittal avatar Mar 15 '22 06:03 gmittal

Related to #114

Michaelvll avatar Mar 15 '22 08:03 Michaelvll

Closing this issue as TPU VM is now supported. https://skypilot.readthedocs.io/en/latest/reference/tpu.html

infwinston avatar Oct 18 '22 04:10 infwinston