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

Key technologies and capabilities for this role

ML InfrastructureData InfrastructureCompute InfrastructureModel TrainingModel ServingDeep Learning FrameworksInference SystemsGPU ComputingComputer VisionPerceptionPredictionPlannerSimulation

Questions & Answers

Common questions about this position

What is the salary for this Principal Software Engineer role?

This information is not specified in the job description.

Is this position remote or does it require working from an office?

This information is not specified in the job description.

What skills are required for this ML Infrastructure Engineer position?

Required skills include experience building and managing large-scale ML infrastructure, excellent leadership skills to lead high-performing teams, strong experience with training frameworks like PyTorch or JAX for GPU-efficient distributed training, experience with GPU-accelerated inference using TensorRT or Ray Serve, and proficiency in Python and/or C++.

What does the team structure look like for this role?

You will lead a team of strong software engineers and act as a force multiplier, while collaborating closely with cross-functional teams including ML researchers, software engineers, data engineers, hardware engineers, and AI teams like Perception, Prediction, and Planner.

What makes a strong candidate for this Principal Software Engineer position?

A strong candidate has experience building large-scale ML infrastructure, demonstrated leadership of high-performing teams, technical expertise in PyTorch/JAX, GPU inference frameworks, and Python/C++, plus the ability to develop strategic vision and mentor engineers.

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