Skip to main content
2022-12-27 Waymo

Senior ML Compiler Engineer, Machine Learning Platform

Mountain View, California, United States
San Francisco, California, United States

Waymo is an autonomous driving technology company with a mission to make it safe and easy for people and things to get where they’re going. Since our start as the Google Self-Driving Car Project in 2009, Waymo has been focused on building the Waymo Driver—The World’s Most Experienced Driver™—to improve everyone's access to mobility while saving thousands of lives now lost to traffic crashes. Our Waymo Driver powers Waymo One, our fully autonomous ride-hailing service, as well as Waymo Via, our trucking and local delivery service. To date, Waymo has driven over 20 million miles autonomously on public roads across 25 U.S. cities and conducted over 20 billion miles of simulation testing.

At Waymo, we are mission-driven and believe deeply in the opportunity of autonomous driving technology to improve mobility and make people's lives better. We are united by purpose and responsibility (for our employees and riders alike). We are looking for kind, committed, employees who have integrity, dream big, work together as one team and create a sense of belonging for one another that is the foundation of our culture. We want each team member to feel welcomed and included in every step of our exciting journey. 

Software Engineering builds the brains of Waymo's fully autonomous driving technology. Our software allows the Waymo Driver to perceive the world around it, make the right decision for every situation, and deliver people safely to their destinations. We think deeply and solve complex technical challenges in areas like robotics, perception, decision-making and deep learning, while collaborating with hardware and systems engineers. If you’re a software engineer or researcher who’s curious and passionate about Level 4 autonomous driving, we'd like to meet you. 

The ML Platform team at Waymo provides a set of tools and technologies to support and automate the lifecycle of the machine learning workflow, including feature and experiment management, model development, debugging & evaluation, deployment, and monitoring. These efforts have resulted in making machine learning more accessible to teams at Waymo, including Perception, Behavior Prediction, Planner, Routing, Maps and Research, ensuring greater degrees of consistency and repeatability, and addressing the “last mile” of getting models into production and managing them once they are in place. We work hand in hand with machine learning experts in all parts of the company as well as our collaborators across Alphabet.

We are looking for engineers with an ML compiler, runtime, or GPU background to help us improve compute performance on our car. You’ll work across the entire ML compiler stack (e.g. TF, XLA, MLIR, and CUDA). You will be pleasantly challenged with deploying a hugely diverse set of Waymo ML models on limited computation resources. Non-exhaustive examples of the types of work you will work on:

  • Collaborate with ML practitioners to understand their models and modify model architectures to run faster on the car.
  • Write CUDA kernels to speed up specific operations.
  • Improve XLA to automatically generate fast code for Waymo models.
  • Build tools to benchmark and analyze workloads.
  • Build tools to train and evaluate models in lower precision.

At a minimum, we’d like you to have:

  • B.Sc in Computer Science, Mathematics or a related field
  • 5+ years of industry experience
  • Strong C++ programming skills
  • Passion for developing and optimizing compilers for modern architectures (MLIR etc.)
  • Working knowledge of compiler architecture front-end and middle-end optimizations, scheduling, register allocation, and back-end code generation

It’s preferred if you have:

  • M.Sc or PhD in Computer Science, Mathematics or a related field
  • Strong Python programming skills
  • Experience with implementing neural networks inference on GPU

The expected base salary range for this full-time position across US locations is listed below. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level. Your recruiter can share more about the specific salary range for the role location or, if the role can be performed remote, the specific salary range for your preferred location, during the hiring process. 

Waymo employees are also eligible to participate in Waymo’s discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements. 

Salary Range
$185,000$221,000 USD

While at Waymo, you will enjoy benefits that cover…

Health and wellness: Our people are at the heart of everything we do. At Waymo, you can enjoy top-notch medical, dental and vision insurance, mental wellness support, a Flexible Spending Account (FSA), a Health Saving Account (HSA), on-site physicians and/or nurses in some locations, and special wellness programs.

Financial wellness: Your financial peace of mind is important to us. At Waymo, we offer competitive compensation, bonus opportunities, equity, a generous 401(k) plan or regional retirement plans, 1-on-1 financial coaching, a 529 College Savings Plan and lots of other perks and employee discounts. 

Flexibility and time off: Take the time you need to relax and recharge. Enjoy the flexibility to work from another location for four weeks per year. We support an on-site, hybrid work model and offer remote working opportunities, paid time off, Waymo recharge days, bereavement, sick, and parental leave.

Supporting families: When it comes to growing your family or caring for your loved ones, you have our full support. Enhanced leave options include paid parental leave (birthing parent gets 24 weeks of paid leave, and non-birthing parent gets 18 weeks of paid leave), and 20 subsidized days of backup childcare or adult/elder care. Access to fertility care or adoption support as you grow your family.

Community and personal development: At Waymo, you’ll find a range of opportunities to grow, connect, and give back. We offer education tuition reimbursement, personal and professional development, mentorship, and other ways to connect through Employee Resource Groups (ERGs), other internal groups, and even time off to volunteer.

Cool perks: Access to Google offices, cafes, wellness centers, personal training sessions, massages, haircuts, bike repairs, office transportation, commuter benefits and so much more. To support your wellbeing at home, you can enjoy at-home fitness and cooking classes, and more. 

* Please note that while our benefits philosophy is the same in every place Waymonauts work, benefits may vary by office/country and are subject to eligibility requirements.

Note: The following is only applicable for roles hired in the US. As we keep the safety of our employees and our communities top of mind, COVID-19 vaccinations continue to be a critical prevention measure to help end the pandemic.  Because of this, Waymo requires all US-based employees who perform work onsite for Waymo to be fully vaccinated against COVID-19. If you are unable to be vaccinated, Waymo will provide a reasonable accommodation consistent with applicable law.

We appreciate your interest in Waymo. Waymo is an equal employment opportunity employer, committed to maintaining a supportive and inclusive workplace for all employees. Waymo does not discriminate against, and prohibits harassment of, any applicant or employee based on race, color, sex, sexual orientation, gender identity, religion, national origin, age, disability, military status, genetic information or any other basis protected by applicable law. Waymo will also consider for employment qualified applicants with criminal records in accordance with applicable law. Waymo is committed to ensuring equal opportunity for qualified individuals with disabilities. If you are an individual with a disability and require an accommodation to participate in the application or interview process, please let the recruiting team know or email waymo-candidatesupport@google.com. (This email address is intended to be used only for requesting accommodations as part of the application process. Other inquiries will not receive a response.)

Apply now

Resume/CV

.pdf, .doc, or .docx only. Attachments must not exceed 10MB.

Cover Letter

.pdf, .doc, or .docx only. Attachments must not exceed 10MB.

Demographic Information (completion voluntary)

You are being given the opportunity to provide the following information in order to help us track and measure the aggregate demographics of our applicants in furtherance of our Equity, Inclusion, and Diversity efforts.  Answering these questions is entirely voluntary, and a decision not to answer will not be held against you.  If you decide to provide demographic information, it will be recorded and maintained in a confidential file.  It will not be considered in the hiring process or as a basis for employment decisions.  Whatever your decision, it will not be considered in the hiring process or as basis for employment.  If you are offered and accept employment with Waymo, the information collected during the application and recruitment process will become part of your confidential employment record.

Gender
Race/Ethnicity
Are you a veteran or reservist?
Do you identify as a member of the LGBTQ+ (e.g., lesbian, gay, bisexual, queer, asexual) community?

PUBLIC BURDEN STATEMENT: According to the Paperwork Reduction Act of 1995 no persons are required to respond to a collection of information unless such collection displays a valid OMB control number. This survey should take about 5 minutes to complete.

Why are you being asked to complete this form?

We are a federal contractor or subcontractor required by law to provide equal employment opportunity to qualified people with disabilities. We are also required to measure our progress toward having at least 7% of our workforce be individuals with disabilities. To do this, we must ask applicants and employees if they have a disability or have ever had a disability. Because a person may become disabled at any time, we ask all of our employees to update their information at least every five years.

Identifying yourself as an individual with a disability is voluntary, and we hope that you will choose to do so. Your answer will be maintained confidentially and not be seen by selecting officials or anyone else involved in making personnel decisions. Completing the form will not negatively impact you in any way, regardless of whether you have self-identified in the past. For more information about this form or the equal employment obligations of federal contractors under Section 503 of the Rehabilitation Act, visit the U.S. Department of Labor’s Office of Federal Contract Compliance Programs (OFCCP) website at www.dol.gov/ofccp.

How do you know if you have a disability?

You are considered to have a disability if you have a physical or mental impairment or medical condition that substantially limits a major life activity, or if you have a history or record of such an impairment or medical condition.

Disabilities include, but are not limited to:

  • Autism
  • Autoimmune disorder, for example, lupus, fibromyalgia, rheumatoid arthritis, or HIV/AIDS
  • Blind or low vision
  • Cancer
  • Cardiovascular or heart disease
  • Celiac disease
  • Cerebral palsy
  • Deaf or hard of hearing
  • Depression or anxiety
  • Diabetes
  • Epilepsy
  • Gastrointestinal disorders, for example, Crohn's Disease, or irritable bowel syndrome
  • Intellectual disability
  • Missing limbs or partially missing limbs
  • Nervous system condition for example, migraine headaches, Parkinson’s disease, or Multiple sclerosis (MS)
  • Psychiatric condition, for example, bipolar disorder, schizophrenia, PTSD, or major depression
Disability Status

If you believe you belong to any of the categories of protected veterans listed below, please indicate by making the appropriate selection. As a government contractor subject to the Vietnam Era Veterans Readjustment Assistance Act (VEVRAA), we request this information in order to measure the effectiveness of the outreach and positive recruitment efforts we undertake pursuant to VEVRAA. Classification of protected categories is as follows:

A "disabled veteran" is one of the following: a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or a person who was discharged or released from active duty because of a service-connected disability.

A "recently separated veteran" means any veteran during the three-year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.

An "active duty wartime or campaign badge veteran" means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.

An "Armed forces service medal veteran" means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.

Veteran Status
Voluntary Self-Identification

For government reporting purposes, we ask candidates to respond to the below self-identification survey. Completion of the form is entirely voluntary. Whatever your decision, it will not be considered in the hiring process or thereafter. Any information that you do provide will be recorded and maintained in a confidential file.

As set forth in Waymo’s Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.

Race
Gender
By submitting my application, I certify that the information provided (including in my resume/CV) is true and correct. I understand that any misrepresentations or false statements made in connection with the recruitment process may be grounds for disqualification from further consideration for employment or, if employed, separation of employment.