Atoms
Senior Machine Learning Engineer
$208K–$264K
San Francisco, CA +1 · Senior · Full-time · Posted Jun 26, 2026
- Location
- San Francisco, CA
San Francisco, California, United States - Experience
- 4+ years of non-internship professional MLE experience.
About the job
Develop machine-learning systems for autonomous transport, including multimodal driving models, model optimization for vehicle hardware, and data pipelines for rare driving scenarios.
What you'll do
- Research and build models for trajectory planning, perception-to-actuation, and interactive world simulation.
- Distill, optimize, profile, and deploy models on vehicle edge hardware.
- Build pipelines to identify and curate rare scenarios from multi-sensor driving datasets for training and simulation.
What they ask for
- 4+ years of non-internship professional MLE experience.
- Expertise applying AI Transformers to robotics, physical actuation, or spatial-temporal data; experience designing or training multimodal systems, large-scale VLA models, or generative Diffusion models.
- Strong Sensor Fusion background combining Cameras, LiDAR, and Radar.
- Fluency in PyTorch or JAX; proficiency in Python and familiarity with C++.
Nice to have
- Experience with multi-task learning, Birds-Eye-View (BEV) frameworks, representation learning, or data tokenization.
- Familiarity with low-level camera/sensor interfaces and robotics hardware.
Pay
$208,000 - $263,500 USD base salary
Tools
PyTorch, JAX, Python, C++, TensorRT, CUDA, Cameras, LiDAR
AI focus
Develop Physical AI for autonomous transport, including reinforcement learning, distillation, multimodal vision-language-action models, interactive world models, edge inference optimization, and data curation for long-tail driving scenarios.
Model-written summary · the original posting is authoritative