Staff Tech Lead, Machine Learning Engineer, Perception jobs in United States
cer-icon
Apply on Employer Site
company-logo

Waymo · 17 hours ago

Staff Tech Lead, Machine Learning Engineer, Perception

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The role involves owning the panoptic segmentation task, collaborating with ML engineering teams, and developing efficient model architectures for occupancy and segmentation.

Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
check
H1B Sponsor Likelynote

Responsibilities

Own the panoptic segmentation task (instance segmentation and semantic segmentation), take responsibility for task scaling and task performance, take ownership of aligning instance segmentation and object detection/tracking (in particular in an end-to-end architecture), create ML methods and recipes to scale and improve tasks
Collaborate with ML engineering/research teams to develop efficient model architectures for representing occupancy and panoptic segmentation, and train and deploy such models with a focus on using large pre-trained models
Collaborate with partner teams to establish efficient human and machine labeling workflows
Analyze panoptic segmentation issues in real-world application, identify issues and root causes, advise or develop short- and long-term solutions
Develop and maintain metrics for 4D (3D + time) instance segmentation and semantic segmentation

Qualification

Machine LearningComputer VisionPythonPyTorchC++ResearchCollaborationProblem Solving

Required

Bachelors in Computer Science or a similar discipline, or an equivalent amount of deep learning experience
7+ years experience in Machine Learning and/or Computer Vision
Experience with Python
Experience with ML frameworks like PyTorch or JAX

Preferred

MS or PhD Degree in Machine Learning, Robotics, Computer Science or a similar discipline
Publications at top-tier conferences like CVPR, ICCV, ECCV, ICLR, ICML, ICRA, IROS, RSS, NeurIPS, AAAI, IJCV, PAMI
Experience with C++

Benefits

Waymo’s discretionary annual bonus program
Equity incentive plan
Generous Company benefits program

Company

Waymo is a mobility technology company that improves transportation by developing self-driving solutions for travelers and daily commuters. It is a sub-organization of Alphabet.

H1B Sponsorship

Waymo 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 (193)
2024 (175)
2023 (268)
2022 (306)
2021 (298)
2020 (317)

Funding

Current Stage
Late Stage
Total Funding
$11.1B
Key Investors
Alphabet
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
2020-05-12Series A· $750M

Leadership Team

leader-logo
Tekedra Mawakana
Co-Chief Executive Officer
linkedin
leader-logo
Elisa de Martel
Chief Financial Officer
linkedin
Company data provided by crunchbase