Community

Home >

Jobs >

Software Engineer, Machine Learning, GPU and TPU Accelerators

California, United States (On-site)

Software Engineer, Machine Learning, GPU and TPU Accelerators

4 Days ago • 3 Years + • Artificial Intelligence

About the job

5 skills required for this role

Add these skills to join the top 1% applicants for this job

cpp

algorithms

data-structures

python

networking

Job Description

This Software Engineer role at Google focuses on Machine Learning (ML) model performance optimization using GPUs and TPUs. Responsibilities include analyzing and improving ML model performance, contributing to performance algorithm research and development, creating automated processes for performance improvements, implementing ML solutions, and collaborating with stakeholders. The ideal candidate will have experience in software development (C++, Python), data structures/algorithms, compilers or accelerators, and ideally experience with post-training quantization and foundational models. The role requires strong collaboration skills and a commitment to efficiency and accuracy.

Must have:

Bachelor's degree or equivalent practical experience
2 years software development (C++, Python)
Data structures/algorithms experience
1 year experience in compilers/accelerators
Analyze and improve ML model performance
Contribute to performance algorithm research

Good to have:

Master's/PhD in CS or related field
Experience with post-training quantization
Experience with foundational models and LLMs
Understanding of ML modeling techniques

Perks:

Bonus
Equity
Benefits

Minimum qualifications:

Bachelor's degree or equivalent practical experience.
2 years of experience in software development (e.g., C++, Python), and with data structures/algorithms.
1 year of experience in compilers, accelerators, or equivalent performance improvements.

Preferred qualifications:

Master's degree or PhD in Computer Science or a related technical field or equivalent practical experience.
Experience with post-training quantization, quantized aware training, or quantized training for ML models.
Experience with foundational models and Large Language Models.
Understanding of the ML modeling techniques and trends.

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. Behind everything our users see online is the architecture built by the Technical Infrastructure team to keep it running. From developing and maintaining our data centers to building the next generation of Google platforms, we make Google's product portfolio possible. We're proud to be our engineers' engineers and love voiding warranties by taking things apart so we can rebuild them. We keep our networks up and running, ensuring our users have the best and fastest experience possible.

The US base salary range for this full-time position is $136,000-$200,000 + bonus + equity + benefits. Our salary ranges are determined by role, level, and location. The range displayed on each job posting reflects the minimum and maximum target salaries for the position across all US locations. Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant education or training. Your recruiter can share more about the specific salary range for your preferred location during the hiring process.

Please note that the compensation details listed in US role postings reflect the base salary only, and do not include bonus, equity, or benefits. Learn more about .

Responsibilities

Analyze and improve performance of Machine Learning (ML) models.
Contribute to research and development of performance algorithms.
Contribute to the creation of automated processes to advance the adoption of performance improvements in Google's Machine Learning (ML).
Implement solutions in ML areas, utilize ML infrastructure, and contribute to model enhancement and data processing.
Collaborate with peers and stakeholders through design and code reviews to ensure standard procedures amongst available technologies (e.g., style guidelines, checking code in, accuracy, testability, and efficiency).

View Full Job Description

$136.0K - $200.0K/yr (Outscal est.)

$168.0K/yr avg.

Sunnyvale, California, United States

Upload your resume, increase your shortlisting chances by 80%

About The Company

Google

2985 Active Jobs

A problem isn't truly solved until it's solved for all. Googlers build products that help create opportunities for everyone, whether down the street or across the globe. Bring your insight, imagination and a healthy disregard for the impossible. Bring everything that makes you unique. Together, we can build for everyone.