Drimva

Google

Senior Software Engineer, Generative AI, Google Research

Sunnyvale, CA, USA · Senior · Full-time · Posted Oct 9, 2026

Experience
5 years of software engineering experience.
Education
Bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, a related technical field, or equivalent practical experience.

About the job

Designs simulation software and AI-driven tools to model AI workloads on future TPU architectures and guide performance and hardware-roadmap decisions.

What you'll do

  • Build scalable performance simulation frameworks and model how AI architectures interact with TPU hardware configurations.
  • Develop AI-agentic tools and optimization engines to automate simulation analyses and identify efficient execution strategies.
  • Produce performance projections and trade-off studies to inform TPU roadmaps and hardware contracts.

What they ask for

  • Bachelor's degree in a related technical field or equivalent practical experience.
  • 5 years of software engineering experience (e.g., C++ and Python).
  • Experience with computer architecture concepts, including memory hierarchies, roofline models, and network topologies.
  • Experience with software design for simulation, performance modeling, or scientific computing.

Nice to have

  • Master's degree or PhD in Computer Science or a related technical field.
  • Experience with LLM-based agentic workflows, performance profiling, architectural simulators, or modern transformer architectures and distributed execution.

Pay

$174000 - $252000 USD base salary

Tools

C++, Python, Mixed-Integer Programming/MIP, JAX, PyTorch/XLA, Pallas

AI focus

Builds AI-driven simulation platforms and LLM-based agentic workflows to automate parameter exploration, data retrieval, and bottleneck diagnosis for AI workload performance and hardware optimization.

Model-written summary · the original posting is authoritative

First seen Oct 21, 2026 · Last verified Nov 5, 2026 (UTC)