Staff Machine Learning Engineer

About the role

Atoms seeks a Staff ML Engineer for physical AI and autonomous transport. Work spans RL trajectory planning, multimodal VLA/world models, sensor fusion, simulated evaluation, and optionally model optimization or massive sensor-data pipelines. Requires 10+ years professional ML engineering, multimodal/transformer models, PyTorch or JAX, Python and C++ familiarity. On-site in San Francisco.

Requirements

• 10+ years non-internship professional ML engineering experience
• Expertise in AI transformers for robotics, physical actuation, or spatiotemporal data
• Track record of multimodal, VLA, or diffusion models
• Sensor fusion across cameras, LiDAR, and radar
• PyTorch or JAX and Python; C++ familiarity
• On-site in San Francisco five days per week

What you'll do

• Develop RL and distillation for trajectory planning
• Prototype multimodal models from visual perception and behavioral goals to vehicle actuation
• Build world models from multi-sensor logs and re-simulate scenarios
• Improve models across novel urban environments and edge cases
• Run simulated validation and regression tests
• Optimize/quantize models for vehicle edge hardware and low-latency inference
• Build data pipelines and active-learning curation for multi-petabyte sensor data

Where this was posted

• LinkedIn

  last seen 2h ago
  View on LinkedIn