Senior Software Engineer, Performance, SOC Architect

Google

Sunnyvale (CA)

On-site

USD 174,000 - 252,000

Full time

12 days ago
Application generator

A complete application in a minute — tailored resume and cover letter, ready to send.

Get past ATS filters

Benefits offered by this job

Equity
Bonus target
Benefits

Job summary

Google is seeking a software engineer to drive next-generation performance and ML-focused workloads. You will analyze trace timelines, memory usage, and ML graph summaries, contributing across the end-to-end stack and custom kernels.

The role emphasizes cross-functional collaboration, system design, and a strong foundation in C++, Python, CUDA, and GPU tooling, with opportunities to influence hardware features and software ecosystems.

Qualifications

  • Bachelor's degree or equivalent practical experience.
  • 5 years of software development experience.
  • 3 years in performance, large-scale data analysis, visualization, or debugging.
  • 3 years in architecture topics including C++, Python, CUDA, GPUs, SoC/ASIC.

Responsibilities

  • Write and test product or system development code.
  • Build understanding of data collection, analysis, and visualization workflows.
  • Support ML paradigms and end-to-end stack analysis tools.
  • Collaborate with leads on model optimization, profiling, and hardware features.
  • Work with Hardware, Driver, Runtime, and Performance Analysis teams.

Skills

Software development
Programming languages
Performance analysis
Large-scale systems
C++
Python
CUDA
GPU programming
System on a Chip
ASIC
Software testing
Software architecture

Education

Bachelor's degree or equivalent practical experience
Master's degree or PhD in CS or related field

Tools

TensorFlow
JAX
PyTorch
XLA

Job description

In most instances, this position requires in-person interviews as part of the hiring process.

Minimum qualifications
  • Bachelor’s degree or equivalent practical experience.
  • 5 years of experience with software development in one or more programming languages.
  • 3 years of experience with performance, large-scale systems data analysis, visualization tools, or debugging.
  • 3 years of experience with Computer Architecture, C++, Python, CUDA, GPU Programming, GPU Drivers, System On a Chip, Application-Specific Integrated Circuit.
  • 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture.
Preferred qualifications
  • Master's degree or PhD in Computer Science or related technical field.
  • 5 years of experience with data structures and algorithms.
  • 1 year of experience in a technical leadership role.
  • Experience with GPU or TPU performance analysis, and a passion for developer productivity.
  • Experience with ML frameworks such as TensorFlow, JAX, and PyTorch, or ML compilers such XLA.
  • Proven track record in open-source software development, including experience in releasing and supporting open-source projects.
About the job
Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to be versatile, display leadership qualities and be enthusiastic to take on new problems across the full-stack as we continue to push technology forward. As a part of this team, you will be focused around driving continuous improvements to the machine learning software/hardware stacks through providing insightful performance debugging for workloads and custom kernels. You will provide insights by summarizing different views of captured profile data such as trace timelines, memory usage, Compiler profiles, ML graph summaries. Individual pay is determined by factors including job-related skills, experience, and relevant education or training. US: $174000 - $252000 (USD) + 15% bonus target + equity + benefits.
Responsibilities
Learn more about benefits at Google .
  • Write and test product or system development code.
  • Learn and build an intuitive understanding of existing data collection, analysis, and visualization workflows.
  • Support new and exciting ML paradigms (such as horizontal scaling for upcoming TPU chips) by making contributions across the end-to-end stack and analysis tools.
  • Partner with Product Area leads to understand model optimization use cases, drive cross functional efforts to deliver on chip profiling requirements, and propose new hardware features.
  • Collaborate across Hardware, Driver, Runtime, and Performance Analysis teams and many other stakeholders.
Google is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. See also Google's EEO Policy and EEO is the Law. If you have a disability or special need that requires accommodation, please let us know by completing our Accommodations for Applicants form .
Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Senior Software Engineer, Performance, SOC Architect
Senior Software Engineer, Performance, SOC Architect

Google Inc. • Sunnyvale (CA)

On-site
USD 174,000 - 252,000
Staff Software Engineer, TPU Performance
Staff Software Engineer, TPU Performance

Google • Sunnyvale (CA)

On-site
USD 207,000 - 300,000
Staff Software Engineer, ML Systems Co-Design
Staff Software Engineer, ML Systems Co-Design

Google • Kirkland (WA)

On-site
USD 207,000 - 300,000
Health insurance
401(k) with company match
Paid time off
+3
Staff Software Engineer, ML Systems Co-Design
Staff Software Engineer, ML Systems Co-Design

Google • Mountain View (CA)

On-site
USD 207,000 - 300,000
Health, dental, vision insurance
401(k) with company match
Paid time off
+4
Tech Lead Manager, Staff Software Engineering, XProf
Tech Lead Manager, Staff Software Engineering, XProf

Google • Sunnyvale (CA)

On-site
USD 207,000 - 301,000
Staff Software Engineer, ML Systems Co-Design
Staff Software Engineer, ML Systems Co-Design

Google Inc. • Sunnyvale (CA)

On-site
USD 207,000 - 300,000
Health, dental, vision, life, and(dis)
401(k) with company match
Paid Time Off: 20 days/year
+4
Software Engineer III, TPU Performance, Hardware and Software Codesign
Software Engineer III, TPU Performance, Hardware and Software Codesign

Google Inc. • Sunnyvale (CA)

On-site
USD 147,000 - 210,000
Senior Software Engineer, Fleet-level ML Performance
Senior Software Engineer, Fleet-level ML Performance

Google • United States

On-site
USD 174,000 - 252,000
Engineering Manager, ML Performance
Engineering Manager, ML Performance

Google • Kirkland (WA)

On-site
USD 207,000 - 300,000
Health insurance
401(k) with company match
Paid Time Off: 20 days per year
+4
Staff Software Engineer, AI/ML, Google Public Sector
Staff Software Engineer, AI/ML, Google Public Sector

Google • Reston (VA)

On-site
USD 207,000 - 300,000