google-deepmind / google-deepmind/xmanager
Support TPU V5e, V6e
Open
- Dominant language
- Python
- Stars
- 921
- Forks
- 69
- Avg merge
- 1d 1h
- Merged PRs (30d)
- 1
Description
Hello,
I use xmanager to launch TPU inference jobs on Vertex AI. Currently xmanager only supports TPU_V2 and TPU_V3:
https://github.com/google-deepmind/xmanager/blob/dd61e339899e8978f1c0cfd845869152b99473bd/xmanager/xm/resources.py#L81
Recently I am having trouble getting quota for these, and it sounds like people are being encouraged to move to TPU_V5E or TPU_V6E. Are there any plans to support these newer TPUs on the roadmap?
Thanks,
Joseph
Contributor guide
Assessment
This issue has not been assessed yet.