Skip to content
ZooxZooxFoster City, CA

Senior/Staff Software Engineer - Planner Frameworks Pipeline

Build and optimize large-scale simulation and ML training pipelines on Ray and Kubernetes to validate autonomous vehicle behavior. Requires 8+ years experience and strong distributed systems background.

219k – 315k/yr
Hybrid8+ YOEML Engineering

About the role

Responsibilities

  • Improve the cost efficiency, reliability, and performance of validation and simulation pipelines
  • Create production-grade APIs, SDKs, and tools to enable varied validations of autonomous behaviors
  • Improve the ML training pipelines supporting the autonomous behavior organization

Requirements

  • Bachelor’s degree in Computer Science or related field
  • 8+ years of industry experience
  • Experience optimizing large-scale distributed systems for cost and efficiency
  • Experience with AWS or similar cloud providers
  • Proficiency with Python and familiarity with C++

Nice-to-Haves

  • Exposure to machine learning workloads (training, inference, data generation) from a cost optimization perspective
  • Background in algorithmic optimization or performance investigation
  • Experience with Ray.io, particularly Ray Core and Ray Data

Skills

PythonC++AWSRay.IoRay CoreRay DataKubernetesDistributed SystemsMachine LearningMl Training Pipelines

Similar roles

ML Engineering jobs
Coinbase

Staff Machine Learning Engineer

CoinbaseUnited States

Staff ML Engineer leading end-to-end identity verification ML systems including document authenticity, face matching, liveness detection, GNN-based identity graphs, and behavioral risk models. Requires 8+ years production ML experience and domain expertise in biometrics or fraud detection.

218k – 257k/yr
Remote8+ YOEML Engineering
Stuut

Member of Technical Staff — Audio and Voice AI

StuutSan Francisco, CA +1

Build and deploy production-grade real-time voice and audio AI systems, including intelligent voice agents, speech-driven workflows, and LLM-powered conversational experiences for financial operations and collections at Stuut. Requires 5+ years software engineering with 2+ years in applied speech/audio AI, experience shipping customer-facing voice systems, and fluency in Python and ML frameworks.

220k – 320k/yr
On-site5+ YOEML Engineering
Perplexity

Member of Technical Staff

PerplexitySan Francisco, CA

Build AI agents that navigate digital environments and perform user tasks. Requires strong AI/ML experience, Python proficiency, and product intuition.

220k – 405k/yr
On-site5+ YOEML Engineering
Perplexity

Member of Technical Staff

PerplexitySan Francisco, CA +1

ML Engineer building and optimizing production recommendation, ranking, and personalization systems that integrate LLMs for Perplexity's AI product.

220k – 405k/yr
On-site5+ YOEML Engineering
Perplexity

Member of Technical Staff

PerplexitySan Francisco, CA

Build and own multimodal AI product and platform systems across the stack at Perplexity. Requires production systems experience, full-stack capability, and strong product judgment.

220k – 405k/yr
On-site5+ YOEML Engineering