Machine Learning, AI, Self Driving Cars, Robotics, Artificial Intelligence.
models can be trained on GPUs such as A100 or H100, we found it was more cost-effective to train them on TPUs. So for training such a model, the scheduler should prefer TPUs when available. This selection should be abstracted away from the user. 6. Scheduling should be decoupled from training