Staff ML Engineer, Perception Research

Why WaymoWorking at Waymo Working at Waymo CultureBenefitsBelongingCommunity EngagementLocationsTeams Teams Software EngineeringOps & Supply ChainAI FoundationsHardware EngineeringProduct & DesignG & AEarly careersHow we hireOpen rolesVisit Waymo.comMy settingsWhy WaymoWorking at WaymoCultureBenefitsBelongingCommunity EngagementLocationsTeamsSoftware EngineeringOps & Supply ChainAI FoundationsHardware EngineeringProduct & DesignG & AEarly careersHow we hireOpen rolesVisit Waymo.comMy settings Kiosk mode Staff ML Engineer, Perception Research Mountain View, California, United States | San Francisco, California, United States | New York City, New York, United States | Kirkland, Washington, United States Full-Time Software Engineering 5285 Apply now Staff ML Engineer, Perception Research 5285 On Site Mountain View, California San Francisco, California New York City, New York Kirkland, Washington Show More Show Less Software Engineering Full-Time View favorites Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver powers Waymo’s fully autonomous ride-hail service and can also be applied to a range of vehicle platforms and product use cases. The Waymo Driver has provided over ten million rider-only trips, enabled by its experience autonomously driving over 100 million miles on public roads and tens of billions in simulation across 15+ U.S. states. The mission of the Waymo AI Foundations team is to develop machine learning solutions addressing open problems in autonomous driving, towards the goal of safely operating Waymo vehicles in dozens of cities and under all driving conditions. As part of our work, we also initiate and foster collaborations with other research teams in Alphabet. AI Foundations areas that we are currently focusing on include reinforcement learning, learning from demonstration, generative modeling, Bayesian inference, hierarchical learning, and robust evaluation. This role follows a hybrid work schedule and reports to a Principal Research Scientist. You Will Conduct comprehensive experimentation to train and deploy state-of-the-art Multimodal LLMs and World models to perform 3D Perception using sensor information from Camera, LiDAR and Radar..Partner effectively with engineering and research teams across Waymo to deploy new models, and implement efficient workflows for model development and continuous training on new front-filled data.Apply and develop techniques such as quantization, pruning, knowledge distillation, and efficient attention mechanisms.Develop and maintain scalable data pipelines for Training & Eval to process data from multiple sources.Design and implement evaluation frameworks for perception models.Develop infrastructure for large-scale model distillation and bulk-inference pipelines for teacher models.Experiment with different model partitioning and sharding strategies to improve scalability and efficiency.Build and maintain tools for performance analysis, profiling (e.g., xprof), and debugging of ML models. You Have PhD or Masters in Computer Science, Machine Learning, Robotics, or a similar technical field, with 4+ years of industry or post-doc research experience in Reinforcement Learning or Foundation Models. Proficiency in implementing model training flows in a scalable, distributed and performant manner such as Data parallel, FSDP and other sharding approaches. Proficiency in JAX, Flax, and potentially TensorFlow/PyTorch.A willingness to work with complexity of globally distributed inference infrastructure.Hands on experience with optimizing the training and inference of Transformer architectures We Prefer PhD in Computer Science, Machine Learning, or Robotics, with a research focus on Reinforcement Learning, Foundation Models, or Multi-Modal learning.Substantial involvement in and contributions to high impact industry AI projects.Experience in generative models for domains such as world models, images, videos, 3D, using techniques such as diffusion or autoregressive models.Experience contributing to frameworks and libraries that improve training speed and scalability (e.g., JAX, Gemax, XManager) Disclosure For WA Based & Remote Roles 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: 401(k) 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 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 $251,000—$310,000 USD We appreciate your interest in Waymo. Waymo is proud to be an equal opportunity employer, committed to creating a culture of belonging and maintaining a supportive workplace for all employees. We welcome applicants of all backgrounds, and employment decisions are based on a candidate’s qualifications, experience, and alignment with job requirements and business needs. 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, family status, pregnancy, 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 making sure our hiring process is accessible for all candidates. If you need assistance applying for a role or participating in the interview process due to a disability, 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.) Video Player is loading. Current Time 0:00 / Duration 2:57 Loaded: 0.00% 0:00 Stream Type LIVE Remaining Time -2:57 1x Chaptersdescriptions off, selectedcaptions settings, opens captions settings dialogcaptions off, selected This is a modal window. Beginning of dialog window. Escape will cancel and close the window. TextColorWhiteBlackRedGreenBlueYellowMagentaCyanOpacityOpaqueSemi-Transparent Text BackgroundColorBlackWhiteRedGreenBlueYellowMagentaCyanOpacityOpaqueSemi-TransparentTransparent Caption Area BackgroundColorBlackWhiteRedGreenBlueYellowMagentaCyanOpacityTransparentSemi-TransparentOpaque Font Size50%75%100%125%150%175%200%300%400% Text Edge StyleNoneRaisedDepressedUniformDrop shadow Font FamilyProportional Sans-SerifMonospace Sans-SerifProportional SerifMonospace SerifCasualScriptSmall Caps End of dialog window. Job Alerts Sign up to receive tailored job alerts delivered straight to your inbox! call-to-action--form#submitFormStart turbo:submit-end->call-to-action--form#submitFormEnd" data-call-to-action--form-target="form" action="https://careers.withwaymo.com/call_to_actions/36ee3fafcbec8a9d447913670f178bc5/form_submissions?job_id=82d1dce7835d3865bce05c9f4098bedb&page_id=f42456e4bd837cabb444c92e64449012" accept-charset="UTF-8" method="post"> First Name (required) 2a6b46d2 Last Name (required) a144956e Email (required) be73b9fc Choose departments and locations you're interested in: 78e0f710 Finalize your job alert by selecting criteria from the dropdowns below. You can select multiple options from each dropdown by returning to the combobox and re-entering the list of options. Submit at the end to create your job alert. Departments Software Engineering Software Engineering Locations Kirkland, Washington, United StatesMountain View, California, United StatesNew York City, New York, United StatesSan Francisco, California, United States Kirkland, Washington, United States Mountain View, California, United States New York City, New York, United States San Francisco, California, United States Not You? By submitting your email address, you agree to our Candidate Privacy Policy. The data you enter will be retained for two years (four years in California), unless you delete it before that time. You may unsubscribe from job alerts or remove your data from the system at any time. Thank you Quick Links How We Hire Our Benefits Related Job Openings There are currently no jobs matching this criteria Some of our Benefits 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, gym membership, and special wellness programs. Financial wellness Your financial peace of mind is important to us. At Waymo, we offer competitive compensation, bonus opportunities, equity, employees provident fund, 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 or hybrid work model and offer remote working opportunities, paid time off, 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 26 weeks of paid leave for birthing parents and 18 weeks of paid leave for non-birthing parents. Community and personal development At Waymo, you'll find a range of opportunities to grow, connect, and give back. We offer education 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, massages, and so much more. To support your wellbeing at home, you can enjoy at-home fitness and cooking classes, and more. 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, gym membership, and special wellness programs. Financial wellness Your financial peace of mind is important to us. At Waymo, we offer competitive compensation, bonus opportunities, equity, employees provident fund, 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 or hybrid work model and offer remote working opportunities, paid time off, 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 26 weeks of paid leave for birthing parents and 18 weeks of paid leave for non-birthing parents. Community and personal development At Waymo, you'll find a range of opportunities to grow, connect, and give back. We offer education 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, massages, and so much more. To support your wellbeing at home, you can enjoy at-home fitness and cooking classes, and more. Learn more Ready to Apply? To ensure thorough consideration of each application, candidates may submit up to 5 job applications within a 60-day period. We encourage you to carefully review job descriptions and apply to positions that best match your skills. Applications beyond this limit will not be considered. call-to-action--form#submitFormStart turbo:submit-end->call-to-action--form#submitFormEnd" data-call-to-action--form-target="form" enctype="multipart/form-data" action="https://careers.withwaymo.com/call_to_actions/4e392923843c0621f469608bfdf96a90/form_submissions?job_id=82d1dce7835d3865bce05c9f4098bedb&page_id=f42456e4bd837cabb444c92e64449012" accept-charset="UTF-8" method="post"> First Name (required) f7f4267f Last Name (required) c5eb6eca Email (required) e6a17598 Phone Number (required) 7fab9bcf 244 results found No results found Afghanistan (+93)Åland Islands (+358)Albania (+355)Algeria (+213)American Samoa (+1)Andorra (+376)Angola (+244)Anguilla (+1)Antigua & Barbuda (+1)Argentina (+54)Armenia (+374)Aruba (+297)Ascension Island (+247)Australia (+61)Austria (+43)Azerbaijan (+994)Bahamas (+1)Bahrain (+973)Bangladesh (+880)Barbados (+1)Belarus (+375)Belgium (+32)Belize (+501)Benin (+229)Bermuda (+1)Bhutan (+975)Bolivia (+591)Bosnia & Herzegovina (+387)Botswana (+267)Brazil (+55)British Indian Ocean Territory (+246)British Virgin Islands (+1)Brunei (+673)Bulgaria (+359)Burkina Faso (+226)Burundi (+257)Cambodia (+855)Cameroon (+237)Canada (+1)Cape Verde (+238)Caribbean Netherlands (+599)Cayman Islands (+1)Central African Republic (+236)Chad (+235)Chile (+56)China (+86)Christmas Island (+61)Cocos (Keeling) Islands (+61)Colombia (+57)Comoros (+269)Congo - Brazzaville (+242)Congo - Kinshasa (+243)Cook Islands (+682)Costa Rica (+506)Côte d’Ivoire (+225)Croatia (+385)Cuba (+53)Curaçao (+599)Cyprus (+357)Czechia (+420)Denmark (+45)Djibouti (+253)Dominica (+1)Dominican Republic (+1)Ecuador (+593)Egypt (+20)El Salvador (+503)Equatorial Guinea (+240)Eritrea (+291)Estonia (+372)Eswatini (+268)Ethiopia (+251)Falkland Islands (Islas Malvinas) (+500)Faroe Islands (+298)Fiji (+679)Finland (+358)France (+33)French Guiana (+594)French Polynesia (+689)Gabon (+241)Gambia (+220)Georgia (+995)Germany (+49)Ghana (+233)Gibraltar (+350)Greece (+30)Greenland (+299)Grenada (+1)Guadeloupe (+590)Guam (+1)Guatemala (+502)Guernsey (+44)Guinea (+224)Guinea-Bissau (+245)Guyana (+592)Haiti (+509)Honduras (+504)Hong Kong (+852)Hungary (+36)Iceland (+354)India (+91)Indonesia (+62)Iran (+98)Iraq (+964)Ireland (+353)Isle of Man (+44)Israel (+972)Italy (+39)Jamaica (+1)Japan (+81)Jersey (+44)Jordan (+962)Kazakhstan (+7)Kenya (+254)Kiribati (+686)Kosovo (+383)Kuwait (+965)Kyrgyzstan (+996)Laos (+856)Latvia (+371)Lebanon (+961)Lesotho (+266)Liberia (+231)Libya (+218)Liechtenstein (+423)Lithuania (+370)Luxembourg (+352)Macao (+853)Madagascar (+261)Malawi (+265)Malaysia (+60)Maldives (+960)Mali (+223)Malta (+356)Marshall Islands (+692)Martinique (+596)Mauritania (+222)Mauritius (+230)Mayotte (+262)Mexico (+52)Micronesia (+691)Moldova (+373)Monaco (+377)Mongolia (+976)Montenegro (+382)Montserrat (+1)Morocco (+212)Mozambique (+258)Myanmar (Burma) (+95)Namibia (+264)Nauru (+674)Nepal (+977)Netherlands (+31)New Caledonia (+687)New Zealand (+64)Nicaragua (+505)Niger (+227)Nigeria (+234)Niue (+683)Norfolk Island (+672)North Korea (+850)North Macedonia (+389)Northern Mariana Islands (+1)Norway (+47)Oman (+968)Pakistan (+92)Palau (+680)Palestine (+970)Panama (+507)Papua New Guinea (+675)Paraguay (+595)Peru (+51)Philippines (+63)Poland (+48)Portugal (+351)Puerto Rico (+1)Qatar (+974)Réunion (+262)Romania (+40)Russia (+7)Rwanda (+250)Samoa (+685)San Marino (+378)São Tomé & Príncipe (+239)Saudi Arabia (+966)Senegal (+221)Serbia (+381)Seychelles (+248)Sierra Leone (+232)Singapore (+65)Sint Maarten (+1)Slovakia (+421)Slovenia (+386)Solomon Islands (+677)Somalia (+252)South Africa (+27)South Korea (+82)South Sudan (+211)Spain (+34)Sri Lanka (+94)St. Barthélemy (+590)St. Helena (+290)St. Kitts & Nevis (+1)St. Lucia (+1)St. Martin (+590)St. Pierre & Miquelon (+508)St. Vincent & Grenadines (+1)Sudan (+249)Suriname (+597)Svalbard & Jan Mayen (+47)Sweden (+46)Switzerland (+41)Syria (+963)Taiwan (+886)Tajikistan (+992)Tanzania (+255)Thailand (+66)Timor-Leste (+670)Togo (+228)Tokelau (+690)Tonga (+676)Trinidad & Tobago (+1)Tunisia (+216)Türkiye (+90)Turkmenistan (+993)Turks & Caicos Islands (+1)Tuvalu (+688)U.S. Virgin Islands (+1)Uganda (+256)Ukraine (+380)United Arab Emirates (+971)United Kingdom (+44)United States (+1) Uruguay (+598)Uzbekistan (+998)Vanuatu (+678)Vatican City (+39)Venezuela (+58)Vietnam (+84)Wallis & Futuna (+681)Western Sahara (+212)Yemen (+967)Zambia (+260)Zimbabwe (+263) Resume (required) d49fa97c Accepted formats: PDF, DOC, DOCX, RTF, ODT, XML, DOT, PNG, GIF, JPG, JPEG, ZIP Cover Letter dae7c18e Accepted formats: PDF, DOC, DOCX, RTF, ODT, XML, DOT, PNG, GIF, JPG, JPEG, ZIP Website 4bb12951 LinkedIn Profile 079eb921 How did you hear about this opportunity? ffe9f509 A personal contact at Google/Alphabet told me about the opportunity A personal contact at Waymo told me about the opportunity Academic Contact (professor recommendation, classmate, etc.) Agency Recruiter Outreach Alphabet's Internal Job Board (Grow) Glassdoor Handshake Indeed LinkedIn Media (blog post, news article, etc.) Social Media (Facebook, Twitter, etc.) University Event (info session, tech talk, luncheon, etc.) University Job Board Waymo Careers Page Waymo Industry Event (conference, career fair, etc.) Waymo Recruiter Outreach Waymo Advertising Word of Mouth Other Work Authorization (required) ff31b1d8 Note: Upon hire, all employees are required to submit proof of legal right to work, consistent with applicable law. Waymo also participates in the E-Verify program. I am authorized to work for any employer in the country in which this position is based. I require, or in the future will require, Waymo's sponsorship to obtain work authorization in the country in which this position is based (e.g. H-1B, TN, etc.) My status to work in the country in which this position is based is unknown. Please review and acknowledge our Candidate Privacy Policy linked below: (required) a76efea4 Waymo Applicant and Candidate Privacy Policy I acknowledge that I have read and understood the terms of the Waymo Applicant and Candidate Privacy Policy. Are you a current or former Alphabet employee, intern, vendor, contractor, or temp (including Google and other Alphabet subsidiaries)? (required) 8e0a3968 Current Alphabet Employee or Intern Former Alphabet Employee or Intern Current or Former member of Alphabet extended workforce Never worked at Alphabet If yes, please provide your LDAP. fbb47d5d Please provide the state/region in which you currently reside. (required) 4bc7a56f New York Illinois Another State in the US APAC EMEA Other Please provide your full legal name as it appears on your government ID (required) 49139cf9 To ensure accuracy of our candidate records (including future employment records should you ultimately join Waymo), please provide your full legal name exactly as it appears on your government-issued ID. To ensure a fair and accurate assessment of each candidate's unique capabilities, Waymo prohibits the use of unauthorized outside assistance during the interview process. This includes, but is not limited to, artificial intelligence (AI) tools, generative software, or third-party resources, unless explicitly authorized by the hiring team. By submitting this application, you acknowledge and agree to adhere to these guidelines. (required) fdd92fb0 I acknowledge the above policies dd533f42 Check this box to join the talent community and sign up for job alerts 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. (required) c1827c83 Not You? By submitting your email address, you agree to our Candidate Privacy Policy. The data you enter will be retained for two years (four years in California), unless you delete it before that time. You may unsubscribe from job alerts or remove your data from the system at any time. Thank you Waymo.comCandidate Privacy PolicyPrivacy PolicyTermsZero ToleranceSafety PublicationsContact UsBlogResearchLegalFirst RespondersWaymo CommunityNotice to CA Residents © 2019-2024 Waymo LLC Search jobs Candidate Details Cookies allow websites to make your Internet experience better. For example, when you set a default language on a website, cookies help the site remember your choice. You can block some or all cookies, but this can cause certain features to stop working. Note that any choice you make here will only affect this website on this browser and device. To learn more about how your data is shared, view our Candidate Privacy Policy page. Manage Consent Preferences Strictly Necessary Cookies These cookies are necessary for the website to function and cannot be switched off in our systems. They are usually only set in response to actions made by you which amount to a request for services, such as setting your privacy preferences , logging in or filling in forms. You can set your browser to block or alert you about these cookies, but some parts of the site will not then work. Performance Cookies These cookies allow us to count visits and traffic sources so we can measure and improve the performance of our site. They help us to know which pages are the most and least popular and see how visitors move around the site. If you do not allow these cookies we will not know when you have visited our site, and will not be able to monitor its performance. Please choose the location(s)