Staff Software Engineer, TPU Performance

Job description

Minimum qualifications: - Bachelor’s degree or equivalent practical experience. - 8 years of experience in software development. - 5 years of experience with one or more of the following: speech/audio (e.g., technology duplicating and responding to the human voice), reinforcement learning (e.g., sequential decision making), ML infrastructure, or specialization in another ML field. - 5 years of experience with ML design and ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning). - 5 years of experience testing, and launching software products, and 3 years of experience with software design and architecture. Preferred qualifications: - Master’s degree or PhD in Engineering, Computer Science, or a related technical field. - 8 years of experience with data structures and algorithms. - Experience with machine learning, compiler optimization, code generation, and runtime systems for GPU architectures (OpenXLA, MLIR, Triton, etc). - Experience in tailoring algorithms and ML models to exploit ML accelerator architecture strengths and minimize weaknesses. - Experience in low-level GPU programming (CUDA, OpenCL, etc.) and performance tuning tech...