Staff Software Engineer, HPC
$201,000 - $315,000 a year
Apply (opens in a new tab)Staff Software Engineer, HPC
Posted 44 days ago
SENIORITY
Lead
SALARY
$201,000 - $315,000 a year
About the role
- Design and implement core services and abstractions for distributed compute infrastructure supporting hundreds of thousands of concurrent jobs
- Work with customer teams and other infrastructure teams to build a multiyear software engineering roadmap for the HPC platform
- Lead multi-quarter, cross team initiatives that drive org-wide improvements
- Create production-grade APIs, SDKs, and tools that make it easy for engineers across Zoox to run large-scale distributed workloads
- Design and improve job scheduling algorithms and auto-scaling policies to maximize reliability and resource availability
- Design multi-region orchestration strategies that optimize for data locality, reliability, and performance
- Identify and resolve systemic reliability and performance issues through profiling, analysis, and collaboration with workload owners across multiple teams
- Evaluate new technologies and paradigms that improve Zoox's computational and storage capabilities
- Develop capacity planning tools and forecasting models to support Zoox's growing compute needs
- Mentor junior engineers, guiding them through their career development
- Experience designing and operating large-scale distributed systems in production
- Experience with Ray.io, particularly Ray Core and Ray Data (or equivalent technologies)
- Experience with Kubernetes, particularly for heterogeneous workloads
- Experience with cloud infrastructure on AWS or similar providers
- Track record of shipping and operating reliable, highly available scalable infrastructure
- Demonstrated ability to prioritize development work and build cross-functional consensus around technical tradeoffs
- Proficiency with PythonBonus Qualifications
- Exposure to machine learning workloads (training, inference, data generation)
- Experience with Kubernetes or SLURM at scale (>10k+ nodes)
- Experience with SLURM workload manager and advanced scheduling policies
- Background in algorithmic optimization or operations research
- Experience building developer tools and platforms used by large engineering organizations$201,000 - $315,000 a year
Before you apply
Applying takes about a minute. These four things decide how fast it moves after that.
Your profile is current
It's what we read first. Occupations, seniority and locations matter more than a long history.
Two examples you can talk through
Not a portfolio — just two pieces of work where you can explain the decisions and what you'd change.
A number in mind
What you're on now and what would make you move. We negotiate better when we know both.
Your notice period
Employers plan around it, and it's the question that stalls offers most often.
Once you apply, someone reads it and calls you before anything reaches the employer — usually within two working days.
