Research Software Engineer, ML Network Optimization, Google Beam jobs in United States
cer-icon
Apply on Employer Site
company-logo

Google · 6 hours ago

Research Software Engineer, ML Network Optimization, Google Beam

Google is a leading technology company that develops next-generation technologies to connect billions of users. The Research Software Engineer will focus on bridging high-level algorithms and low-level hardware performance, working on projects critical to Google’s needs while collaborating with researchers and engineers to push the boundaries of communication technology.

AppsArtificial Intelligence (AI)Cloud StorageSearch EngineSEO
check
Growth Opportunities
check
H1B Sponsor Likelynote

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 structuresAlgorithmsKernel programmingTechnical leadershipBachelor’s degreeMaster's degreePhDAccessible technologies

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 specializes in internet-related services and products, including search, advertising, and software. It is a sub-organization of Alphabet.

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

leader-logo
Sundar Pichai
CEO
linkedin
leader-logo
Thomas Kurian
CEO - Google Cloud
linkedin
Company data provided by crunchbase