Principal Software Engineer, ML Infrastructure at Zoox

Foster City, California, United States

Apply Now

Not SpecifiedCompensation

Expert & Leadership (9+ years)Experience Level

Full TimeJob Type

UnknownVisa

Autonomous Vehicles, Transportation, AIIndustries

Requirements

Experience building and managing large-scale ML infrastructure that powers the development of large-scale ML models
Excellent leadership skills with a demonstrated ability to lead high-performing engineering teams
Strong experience with training frameworks like PyTorch, JAX, etc., leveraging GPUs efficiently for distributed model training
Experience with GPU-accelerated inference using TensorRT, Ray Serve, or similar frameworks
Proficient in Python and/or C++

Responsibilities

Develop and execute a strategic vision for ML Infrastructure that will unlock innovation in autonomous driving and enhance rider experience
Lead the design and implementation of cutting-edge infrastructure spanning all stages of an ML lifecycle from data preparation to training to evaluation, deployment, and serving
Collaborate closely with cross-functional teams, including ML researchers, software engineers, data engineers, and hardware engineers, to define requirements and align on architectural decisions
Lead the design and development of Data, Compute, Model Training, and Serving Infrastructure
Work across all AI teams within Zoox, including Perception, Prediction, Planner, Simulation, Collision Avoidance
Lead a team of strong software engineers and act as a force multiplier for teams
Enable the engineers in the team to grow their careers by providing technical guidance and mentorship

Skills

ML Infrastructure

Data Infrastructure

Compute Infrastructure

Model Training

Model Serving

Deep Learning Frameworks

Inference Systems

GPU Computing

Computer Vision

Perception

Prediction

Planner

Simulation

Zoox

Develops autonomous electric ride-hailing vehicles

About Zoox

Zoox focuses on creating a fully integrated autonomous ride-hailing service designed specifically for urban transportation. Unlike other companies that modify existing vehicles for self-driving capabilities, Zoox has engineered its own vehicle from the ground up, optimizing it for safety, efficiency, and passenger comfort. The vehicle features advanced sensors, cameras, and AI systems that allow it to navigate complex city environments. It can carry up to four passengers in a spacious interior, ensuring a pleasant ride experience. Safety is paramount, with rigorous testing and collaboration with regulatory bodies to meet high standards. Additionally, Zoox's vehicles are fully electric, promoting sustainability by reducing emissions. The company also develops a user-friendly ride-hailing platform that uses data analytics to improve routing and reduce wait times. Zoox aims to enhance urban mobility while prioritizing safety and environmental responsibility.

Menlo Park, CaliforniaHeadquarters

2014Year Founded

$1,206.2MTotal Funding

ACQUISITIONCompany Stage

Automotive & Transportation, Energy, AI & Machine LearningIndustries

1,001-5,000Employees

Benefits

Health Insurance

Maternity & Paternity Leave

Vacation & Paid Time Off

Sick Days

Free Lunch or Snacks

Employee Assitance Program

Risks

Increased competition from Waymo and Cruise may impact Zoox's market share.

Expansion into new cities introduces regulatory and operational challenges for Zoox.

Partnership with Williams Racing may divert focus from core vehicle development.

Differentiation

Zoox designs vehicles specifically for autonomous operation, unlike retrofitting existing cars.

Zoox's vehicles feature a unique, futuristic design with advanced AI systems.

Zoox offers a fully integrated ride-hailing service optimized for safety and efficiency.

Upsides

Zoox's expansion into Austin and Miami opens new markets and data collection opportunities.

Partnership with Williams Racing boosts Zoox's brand visibility in the U.S. market.

Hiring former Tesla executive Gao could accelerate Zoox's technological advancements.

Land your dream remote job 3x faster with AI

Try Jobo Free