Zoox Logo

Zoox

Senior/Staff Software Engineer - Machine Learning & System Optimization

Reposted 15 Days Ago
Be an Early Applicant
Hybrid
Seattle, WA
226K-307K Annually
Senior level
Hybrid
Seattle, WA
226K-307K Annually
Senior level
Design and optimize on-vehicle perception inference: allocate CPU/GPU resources, compress and quantize large multimodal models, build TensorRT model conversion pipelines, and deliver production-quality low-latency C++/CUDA inference for power- and thermal-constrained vehicle SoCs.
The summary above was generated by AI

The Perception team is pioneering the development of a multi-modality foundation model to drive the next generation of autonomous system intelligence.

As a Machine Learning and System Optimization Engineer, you will orchestrate and allocate overall system capacity to various core perception models running on-bot, as well as drive large initiatives that allow for more efficient inference by sharing various parts of the perception stack with one another.

You will focus on bringing highly efficient, production-ready large-scale models to our on-vehicle stack. We are looking for experts with hands-on experience compressing, accelerating, and deploying complex models, including LLMs, VLMs, or foundation models, for power- and thermal-constrained vehicle SoCs.

In addition, you will optimize ML models, write custom CUDA kernels, and build highly concurrent inference code to ensure real-time, deterministic execution on edge devices.

In this role, you will:

  • Allocate and distribute system resources (CPU/GPU/interconnect) to various models and inference engines running on the robot.

  • Spearhead cross-cutting initiatives that allow for better compute utilization through sharing/fusing models and better scheduling strategies.

  • Optimize large-scale models (Multi-Modal Sensor Fusion models, LLMs, VLMs) using advanced quantization (PTQ, QAT), pruning, mixed-precision inference frameworks, and parameter-efficient fine-tuning (LoRA, QLoRA).

  • Architect and implement model conversion and compilation pipelines using TensorRT for edge deployment.

  • Write production-level, low-latency, and memory-safe C++ and CUDA code for real-time inference on vehicle systems.

Qualifications:

  • Deep experience in system and performance optimization in CPU/GPU systems designed for low latency or high throughput.

  • Deep expertise in working with real-time systems & required constraints such as processing latency, memory utilization, and memory bandwidth pressure.

  • Deep expertise in model quantization (PTQ, QAT) and mixed-precision inference frameworks (INT8, FP8, FP4, BF16/FP16).

  • Proficiency in low-level programming for AI accelerators, specifically developing and optimizing custom ML OPs and TensorRT Plugins with efficient CUDA kernel implementations.

  • Production-level C++ (14/17/20) and Python programming skills, with experience developing concurrent, memory-safe, real-time inference code for edge devices.

Bonus Qualifications:

  • Prior experience in high-performance robotics applications such as AV/drones/robots.

  • Familiarity with SOTA autonomous driving perception algorithms (temporal 3D object detection, BEV, 3D Occupancy Networks) and multi-modal sensor processing (Vision, LiDAR, Radar).

  • Experience with end-to-end autonomous driving paradigms (VLM/VLA models, Foundation models) and edge deployment technologies (e.g., TensorRT-LLM).

About Zoox
Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. Sitting at the intersection of robotics, machine learning, and design, Zoox aims to provide the next generation of mobility-as-a-service in urban environments. We’re looking for top talent that shares our passion and wants to be part of a fast-moving and highly execution-oriented team.

Follow us on LinkedIn

Accommodations
If you need an accommodation to participate in the application or interview process please reach out to [email protected] or your assigned recruiter.

A Final Note:
You do not need to match every listed expectation to apply for this position. Here at Zoox, we know that diverse perspectives foster the innovation we need to be successful, and we are committed to building a team that encompasses a variety of backgrounds, experiences, and skills.

Similar Jobs

4 Minutes Ago
Hybrid
167K-291K Annually
Senior level
167K-291K Annually
Senior level
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Own and drive the Microsoft-ServiceNow product partnership end-to-end: technical due diligence, dual-roadmap alignment, integration delivery, partnership positioning, cross-functional coordination, and measuring adoption and GTM impact to guide roadmap and investment decisions.
Top Skills: Microsoft AiAzureServicenow
5 Minutes Ago
Hybrid
261K-332K Annually
Senior level
261K-332K Annually
Senior level
Consumer Web • Healthtech • Professional Services • Social Impact • Software
Lead and grow the Web Platform engineering team through people development, strategic planning, and cross-functional partnership. Define and advocate platform roadmap, drive platform investments that enable product goals, align multiple product teams, and ensure visibility through documentation, metrics, and stakeholder engagement.
Top Skills: ClaudeCursorDatadogEddyGeminiGitNext.JsPagerdutyReactRemixSentryTypescript
20 Minutes Ago
Remote or Hybrid
United States
130K-160K Annually
Senior level
130K-160K Annually
Senior level
Enterprise Web • HR Tech • Information Technology • Software • Cybersecurity
Serve as a trusted, embedded cybersecurity consultant across the customer lifecycle: lead technical pre-sales demonstrations and POVs, advise C-level and technical stakeholders, design cyber workforce resilience programs, drive platform adoption, operationalize metrics and reports, act as escalation between sales/product, and influence product innovation to maximize value from the Immersive One platform.
Top Skills: Anti-MalwareDatabasesFirewallsIdsImmersive OneLinuxNetworkingPublic CloudSIEMWindows

What you need to know about the Chicago Tech Scene

With vibrant neighborhoods, great food and more affordable housing than either coast, Chicago might be the most liveable major tech hub. It is the birthplace of modern commodities and futures trading, a national hub for logistics and commerce, and home to the American Medical Association and the American Bar Association. This diverse blend of industry influences has helped Chicago emerge as a major player in verticals like fintech, biotechnology, legal tech, e-commerce and logistics technology. It’s also a major hiring center for tech companies on both coasts.

Key Facts About Chicago Tech

  • Number of Tech Workers: 245,800; 5.2% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: McDonald’s, John Deere, Boeing, Morningstar
  • Key Industries: Artificial intelligence, biotechnology, fintech, software, logistics technology
  • Funding Landscape: $2.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Pritzker Group Venture Capital, Arch Venture Partners, MATH Venture Partners, Jump Capital, Hyde Park Venture Partners
  • Research Centers and Universities: Northwestern University, University of Chicago, University of Illinois Urbana-Champaign, Illinois Institute of Technology, Argonne National Laboratory, Fermi National Accelerator Laboratory

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account