Apply on Employer Site

Google · 3 hours ago

Research Software Engineer, ML Network Optimization, Google Beam

San Francisco, CA

Full-time

Hybrid

Senior Level

$166K/yr - $244K/yr

5+ years exp

Google is a leading technology company known for its innovative products and services. As a Research Software Engineer, you will join a creative team focused on advancing technologies in machine learning and graphics, working to bridge the gap between high-level algorithms and low-level hardware performance.

AppsArtificial Intelligence (AI)Cloud StorageSearch EngineSEO

Growth Opportunities

H1B Sponsor Likely

Responsibilities

Research and implement emerging best practices in network design (e.g., quantization, distillation) to accelerate network performance without loss of quality

Lead low-level performance optimization efforts and write fast custom kernels to reduce per-call GPU resource requirements and latency

Analyze and profile complex ML models and graphics pipelines to identify and resolve performance bottlenecks

Collaborate with hardware and compiler teams to influence next-generation architecture and software stack requirements

Navigate hardware-software co-design to facilitate long-term gains in the performance of our ML software and hardware stack

Qualification

C++PythonMachine Learning OptimizationML infrastructureData structuresAlgorithmsGPU/TPU architectureKernel programmingCompiler technologiesAccessible technologiesTechnical leadership

Required

Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience

5 years of experience with software development in one or more programming languages, including C++ and Python

3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture

3 years of experience with one or more of the following: Machine Learning Optimization (e.g., quantization, distillation), GPU/TPU hardware architecture, Kernel programming, or specialization in another ML field

3 years of experience with ML infrastructure (e.g., model deployment, model evaluation, optimization, data processing, debugging)

Preferred

Master's degree or PhD in Computer Science, Computer Engineering, or a related technical field

5 years of experience with data structures and algorithms, and hardware-software co-design

1 year of experience in a technical leadership role or driving architectural transitions

Experience with kernel programming (e.g., CUDA, OpenCL, Vulkan, Triton), compiler technologies (e.g., XLA), and optimizing real-time media pipelines

Experience developing accessible technologies

Benefits

Bonus

Equity

Benefits

Company

Google

Glassdoor4.4

Google specializes in internet-related services and products, including search, advertising, and software. It is a sub-organization of Alphabet.

Founded in 1998

Mountain View, California, USA

10001+ employees

https://www.google.com

H1B Sponsorship

Google has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (8763)

2024 (8872)

2023 (9682)

2022 (11626)

2021 (9109)

2020 (9785)

Funding

Current Stage

Public Company

Total Funding

$26.1M

Key Investors

Andy Bechtolsheim

2004-08-19IPO

1999-06-07Series Unknown· $25M

1998-11-01Angel· $1M

Leadership Team

Sundar Pichai

CEO

Thomas Kurian

CEO - Google Cloud

Recent News

The Indian Express

Who is Ragini Das? Rejected by Google 12 years ago, she now leads its startup arm in India

2025-10-08

Tech.eu

Profits up at Google UK, but nearly 400 jobs cut

2025-10-04