Skip to content
Zoox

Machine Learning Engineer - 3D Sensor Simulation

Develops and optimizes 3D sensor simulations using GenAI/ML and graphics techniques for realistic camera, lidar, and radar data in autonomous vehicles. Requires 2+ years in neural rendering like NeRFs/Gaussian Splatting, PyTorch/TensorFlow, Python/C++.

173k – 244kSeattle, WAML EngineeringHybrid2+ YOE
Apply

About the role

Responsibilities

  • Research, implement, and optimize state-of-the-art 3D rendering of sensor data, leveraging GenAI/ML and 3D graphics.
  • Develop realism metrics with V&V to show measurable impact of improved sensor fidelity.
  • Collaborate with Perception and Safety teams to improve realism of sensor simulation for high fidelity synthetic data.
  • Improve rendering and ML inference tooling for generating realistic data at scale.

Qualifications

  • 2+ years of industry experience, and/or PhD, developing neural rendering techniques like Gaussian Splatting, NeRFs, or 3D reconstruction.
  • 2+ years of experience developing software with Python and/or modern C++.
  • Expertise with machine learning frameworks PyTorch or Tensorflow.
  • Familiarity with 3D graphics algorithms, such as 3D geometry and camera models.
  • Strong mathematical skills and understanding of 3D linear algebra and probabilistic techniques.

Bonus Qualifications

  • Master's or PhD in computer science, mathematics, physics, or related field.
  • Experience with generative models for 3D content pipelines using applications like Houdini, Maya, or Blender.
  • Experience in 3D rendering for simulation, games, cloud computing, or VFX.

Skills

PyTorchTensorFlowPythonC++Gaussian SplattingNerfs3D Reconstruction3D GraphicsNeural RenderingLinear Algebra

Similar roles

ML Engineering jobs
Mirage

ML Engineer, Generative Video

Build and scale video generation models at Mirage. Optimize large-scale multimodal models for training efficiency, low-latency inference, and production deployment using PyTorch, CUDA, and distributed systems. 2+ years industry experience required; on-site in NYC.

175k – 275kNew York, NYML EngineeringOn-site2+ YOECUDAFsdp
Glean

Machine Learning Engineer, Infrastructure

Build and improve ML systems and data pipelines infrastructure to support modeling engineers. Requires 2+ years experience, bachelor's in CS/math/sciences, strong coding in Python/Go/Java/C++, and production ML infra expertise.

175k – 270kUnited StatesML EngineeringHybrid2+ YOEGoC++
Convex

Software Engineer, AI

Convex is seeking a Software Engineer to build and architect their AI platform, focusing on creating seamless agentic development experiences. This role involves working across APIs, plugins, and AI framework integrations, identifying gaps, and prioritizing projects to enhance the platform.

170k+San Francisco, CAML EngineeringOn-siteEntry levelAPIsTypeScript
Figma

Data Scientist, Core Data - PhD (2026)

Research-minded Data Scientist building experimentation platforms, ML analytical systems, and causal inference models for data-driven decisions at Figma. Requires PhD in quantitative field, SQL/Python fluency, and strong stats/ML foundation.

170k – 178kNew York, NY +1ML EngineeringRemoteEntry levelRSQL
Zoox

Machine Learning Engineer - 3D Sensor Simulation

Develops and optimizes ML-driven 3D sensor simulations (cameras, lidar, radar) using GenAI and graphics techniques to generate realistic synthetic data for AV testing. Requires 2+ years experience with neural rendering (NeRFs, Gaussian Splatting), PyTorch/TensorFlow, Python/C++, and 3D math.

176k – 257kFoster City, CAML EngineeringHybrid2+ YOEC++Nerf