Senior Software Engineer & SOC Architect - GPU/ML

Google; Sunnyvale (CA); On-site; USD 174,000 - 252,000; Full time

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

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.