remove vertex-ai 8-core limitation
remove vertex-ai 8-core limitation

PiperOrigin-RevId: 454192899
Change-Id: I29844b0d7a357c4496c0ba24e8ec70d05b448fde
GitOrigin-RevId: b6dc36ff0ac65ac3c1fd5a0ee842a449ed697978

deepmind/xmanager,deepmind/xmanager