Staff Software Engineer — Scalable HPC Platform Lead

Zoox; Foster City (CA); On-site; USD 180,000 - 260,000; Full time

Job summary

Zoox is seeking an experienced Staff Software Engineer to build and operate a high-performance computing (HPC) platform that scales with our autonomous vehicle development. You will modernize our HPC stack using Ray.io, SLURM, and Kubernetes, ensuring reliability and world-class developer velocity across teams from data engineering to AI model training.

You will define Zoox's HPC strategy, work with Autonomy and Software stakeholders, and lead cross-team initiatives to improve compute throughput

Qualifications

• Experience designing and operating large-scale distributed systems in production.
• Proficiency with Python and distributed compute workloads.
• Experience with Ray.io and Kubernetes.
• Strong scheduling/scaling knowledge and optimization mindset.
• Cloud infrastructure experience (AWS or similar).
• Ability to mentor and lead cross-functional teams.

Responsibilities

• Design and implement core services for distributed compute infrastructure supporting hundreds of thousands of concurrent jobs.
• Work with customer teams and other infrastructure groups to build a multi-year HPC roadmap.
• Lead multi-quarter, cross-team initiatives driving org-wide improvements.
• Create production-grade APIs, SDKs, and tools for Zoox engineers to run large-scale workloads.
• Design and improve job scheduling algorithms and auto-scaling policies for reliability.
• Develop multi-region orchestration strategies optimizing data locality and performance.
• Identify and resolve systemic reliability and performance issues via profiling and collaboration.
• Evaluate new technologies to enhance compute and storage capabilities.
• Develop capacity planning tools and forecasting models for growing compute needs.
• Mentor junior engineers and guide career development.

Skills

Distributed systems

Ray.io

Kubernetes

Python

AWS

Job scheduling

Capacity planning

Mentoring

Performance optimization

Orchestration

Tools

Ray Core

Ray Data

SLURM

Kubernetes

Job description

Zoox is seeking an experienced Staff Software Engineer to build and operate a high-performance computing (HPC) platform that scales with our autonomous vehicle development. You will modernize our HPC stack using Ray.io, SLURM, and Kubernetes, ensuring reliability and world-class developer velocity across teams from data engineering to AI model training.

You will define Zoox's HPC strategy, work with Autonomy and Software stakeholders, and lead cross-team initiatives to improve compute throughput