Staff Software Engineer, Machine Learning Compilers, Edge TPU

Staff Software Engineer Machine Learning Compilers Edge TPU - Google Careers

Careers Skip navigation links home Home work_outline Jobs noogler_hat Students google How we work handyman How we hire person_outline Your career help_outline Help link feedback Send feedback more_vert Help Send Feedback Sign in

Careers home Home work_outline Jobs noogler_hat Students google How we work handyman How we hire person_outline Your career

job details arrow_back Back to jobs Jobs search results 3688 jobs matched Global Solutions Lead Go-To-Market Atlanta GA USA Boulder CO USA 7 more 6 more AI Sales Specialist Google Cloud Canberra ACT Australia Senior Analyst Mandiant Threat Intelligence English Finnish Finland | Remote eligible Partner Development Manager Global System Integrator Japanese Tokyo Japan Product Quality Engineer Hardware Google Cloud Sunnyvale CA USA Technical Program Manager II Network Planning Cloud Supply Chain Sunnyvale CA USA Head of HR Technology Transformation Office Sunnyvale CA USA New York NY USA Software Engineer III Performance Google Cloud Platforms Sunnyvale CA USA Account Executive New Business Sales Italian English Dublin Ireland Senior Director UX YouTube Emerging Experience and Community San Bruno CA USA Administrative Business Partner Developer and Experiences Sunnyvale CA USA Software Engineer III Engineering Productivity Google Cloud Kirkland WA USA Investigator Extended Workforce Investigations Austin TX USA Ann Arbor MI USA 9 more 8 more Engineering Analyst Trust and Safety AI Safety Kirkland WA USA Product Manager Events Platform Global Business Organization Engineering London UK Creator Experience Senior Policy Analyst YouTube San Bruno CA USA Product Support Manager Consumer Payments gUP Singapore Account Executive New Business Sales Central Europe English Greek Hamburg Germany Staff Software Engineering Google Distributed Cloud Hosted Bengaluru Karnataka India Software Engineer Data Center Infrastructure Management Cooling Sunnyvale CA USA

Showing 1 to 20 of 3688 rows 1‑20 of 3688 navigate_next

Follow Life at Google on More about us About us open_in_new Contact us open_in_new Press open_in_new Related Information Investor relations open_in_new Blog open_in_new Equal Opportunity Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve creating a culture of belonging and providing an equal employment opportunity regardless of race creed color religion gender sexual orientation gender identity expression national origin disability age genetic information veteran status marital status pregnancy or related condition including breastfeeding expecting or parents-to-be criminal histories consistent with legal requirements or any other basis protected by law. See also Googles EEO Policy Know your rights workplace discrimination is illegal Belonging at Google and How we hire.

More about us expand_more Related information expand_more Equal opportunity expand_more Privacy open_in_new Applicant & Candidate Privacy open_in_new Terms open_in_new Manage cookies help Help open_in_new arrow_back Back to jobs search

Staff Software Engineer Machine Learning Compilers Edge TPU share link Copy link email Email a friend corporate_fare Google place Mountain View CA USA Kirkland WA USA bar_chart Advanced Advanced Experience owning outcomes and decision making solving ambiguous problems and influencing stakeholders deep expertise in domain. Apply share link Copy link email Email a friend info_outline X In accordance with Washington state law we are highlighting our comprehensive benefits package which is available to all eligible US based employees. Benefits for this role include:

• Health dental vision life disability insurance
• Retirement Benefits 401k with company match
• Paid Time Off 20 days of vacation per year accruing at a rate of 6.15 hours per pay period for the first five years of employment
• Sick Time 40 hours year statutory where applicable 5 days
• event discretionary
• Maternity Leave Short-Term Disability Baby Bonding 28-30 weeks
• Baby Bonding Leave 18 weeks
• Holidays 13 paid days per year.

Note: By applying to this position you will have an opportunity to share your preferred working location from the following: Mountain View CA USA Kirkland WA USA.

Minimum qualifications

• Bachelors degree or equivalent practical experience.
• 8 years of experience in software development.
• 5 years of experience testing and launching software products and 3 years of experience with software design and architecture.
• 5 years of experience with Machine Learning compilers optimization parallelization etc.
• 5 years of experience with relevant ML design and ML infrastructure e.g. model deployment model evaluation etc.

Preferred qualifications

• Masters degree or PhD in Engineering Computer Science or a related technical field.
• Experience in optimizing ML models for inference.
• Experience compiling for heterogeneous architectures across IPs including CPU GPU and NPUs.
• Experience with hardware-software co-design.
• Experience in MLIR or Low Level Virtual Machine LLVM.
• Experience in compiler development particularly in the context of accelerator-based architectures vector instruction optimizations or vectorizing compilers.

About the job Googles 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. Were 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 Googles 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.

EdgeTPU is a family of embedded machine learning ML accelerators that aim toward a broad set of applications. The compiler team is responsible for analysis optimization and compilation of ML models focusing EdgeTPU.

Googles mission is to organize the worlds information and make it universally accessible and useful. Our team combines the best of Google AI Software and Hardware to create radically helpful experiences. We research design and develop new technologies and hardware to make computing faster seamless and more powerful. We aim to make peoples lives better through technology.

The US base salary range for this full-time position is 207000-300000 bonus equity benefits. Our salary ranges are determined by role level and location. 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 benefits at Google.

Responsibilities

• Work as part of the EdgeTPU compiler team building ML compilers for EdgeTPU hardware and analyzing and improving the compiler quality and performance on optimization decisions correctness and compilation time.
• Work with and extend ML authoring frameworks including JAX Pytorch to compile ML models for the EdgeTPU.
• Work with ML runtime systems to deploy optimized ML models on the EdgeTPU.
• Work with EdgeTPU architects to design the HardwareSoftware HWSW interface and co-optimizations between CPU GPU and TPU.
• Collaborate with ML model developers researchers and EdgeTPU hardwaresoftware teams to accelerate the transition from research ideas to user experiences running on the EdgeTPU.

Information collected and processed as part of your Google Careers profile and any job applications you choose to submit is subject to Googles Applicant and Candidate Privacy Policy.

Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve creating a culture of belonging and providing an equal employment opportunity regardless of race creed color religion gender sexual orientation gender identity expression national origin disability age genetic information veteran status marital status pregnancy or related condition including breastfeeding expecting or parents-to-be criminal histories consistent with legal requirements or any other basis protected by law. See also Googles EEO Policy Know your rights workplace discrimination is illegal Belonging at Google and How we hire.

If you have a need that requires accommodation please let us know by completing our Accommodations for Applicants form.

Google is a global company and in order to facilitate efficient collaboration and communication globally English proficiency is a requirement for all roles unless stated otherwise in the job posting.

To all recruitment agencies Google does not accept agency resumes. Please do not forward resumes to our jobs alias Google employees or any other organization location. Google is not responsible for any fees related to unsolicited resumes.

Follow Life at Google on More about us About us open_in_new Contact us open_in_new Press open_in_new Related Information Investor relations open_in_new Blog open_in_new Equal Opportunity Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve creating a culture of belonging and providing an equal employment opportunity regardless of race creed color religion gender sexual orientation gender identity expression national origin disability age genetic information veteran status marital status pregnancy or related condition including breastfeeding expecting or parents-to-be criminal histories consistent with legal requirements or any other basis protected by law. See also Googles EEO Policy Know your rights workplace discrimination is illegal Belonging at Google and How we hire.

More about us expand_more Related information expand_more Equal opportunity expand_more Privacy open_in_new Applicant & Candidate Privacy open_in_new Terms open_in_new Manage cookies help Help open_in_new Google apps Main menu