← All roles

Senior Software Engineer, Generative AI, Google Research

Google
Apply on company site
Sunnyvale, CA, USA$174k - $252kFull-timePosted 2w ago

About the role

Software engineer role focused on design and development of performance simulation frameworks and AI-driven platforms to model AI workloads on next-generation hardware, with emphasis on TPU configurations, performance analysis, and agentic workflows.

Responsibilities

  • Design, build, and maintain scalable performance simulation frameworks to accurately evaluate training and serving metrics for next-generation TPU configurations.
  • Model how future model architectures interact with architectural variables (e.g., SRAM capacity/bandwidth, HBM contention, ICI network topologies, and live failover mechanisms).
  • Develop and integrate AI-agentic platforms and toolchains to automate simulator sweep setup, parameter sensitivity analyses, and data extraction over gigabytes of simulation sweeps—reducing analysis cycles from weeks to hours.
  • Build analytical and optimization engines (e.g., Mixed-Integer Programming/MIP, simulated annealing, roofline estimation) to identify Pareto-optimal execution strategies (sharding, tensor placement, weight pinning, prefetching).
  • Formulate clear, data-grounded performance projections and comparative studies to inform multi-year TPU platform roadmaps and hardware contracts.

Qualifications

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

Skills

Required:C++Python

Benefits

  • Learn more about benefits at Google (link in job description).
Last updated Oct 23, 2026. Interested? Apply on the company site.

About Google

Google Search homepage

llmtext-to-speechvoice-ai-stack
Company profileWebsite

More open roles at Google