GenAI SoC Architect — ML Hardware & Systems Innovation

Google; San Diego (CA); On-site; USD 192,000 - 278,000; Full time

Benefits offered by this job

Equity

Job summary

Google is seeking a seasoned hardware architect to lead GenAI ML architecture exploration for Tensor mobile SoCs. You will collaborate with research, system architecture, and compiler teams to drive future workloads across hardware and software.

The role requires deep expertise in AI model architectures and proficiency in C/C++/Python, with emphasis on power and performance. Individual pay is determined by factors including skills and education.

Qualifications

• Bachelor's degree in Computer Science, Electrical Engineering or a related field, or equivalent practical experience.
• 10 years of work or academic research experience in computer or chip architecture, performance, or compiler.
• Experience with modern Generative AI model architectures (e.g., Large Language Models, Vision Transformers, Image Diffusion Models, etc.).
• Experience with one or more general purpose programming languages (e.g., C, C++, or Python) and deep learning frameworks (e.g., TensorFlow, JAX, or PyTorch).

Responsibilities

• Drive GenAI ML architecture exploration for Tensor mobile SoCs with researchers and engineers to optimize future workloads across the tech stack.
• Define system architecture requirements for future Generative AI use cases with PM teams.
• Apply advanced research to achieve power and performance gains on Generative AI workloads.
• Optimize model scheduling on the TPU compute engines to improve GenAI performance.

Skills

Generative AI architectures

C/C++/Python

Deep learning frameworks

Education

Bachelor's degree in CS/EE or related field

Master's or PhD in CS/EE

Job description

Google is seeking a seasoned hardware architect to lead GenAI ML architecture exploration for Tensor mobile SoCs. You will collaborate with research, system architecture, and compiler teams to drive future workloads across hardware and software.

The role requires deep expertise in AI model architectures and proficiency in C/C++/Python, with emphasis on power and performance. Individual pay is determined by factors including skills and education.