Senior Staff Software Engineer, ML Performance, GPU at Google — JAOS