Staff ML Engineer — Physical AI for Real-World Robotics

ATOMS Careers page; San Francisco (CA); On-site; USD 273,000 - 345,000; Full time

Benefits offered by this job

Medical, Dental, Vision Insurance

401(k)

Unlimited Flexible Time Off

Job summary

ATOMS Careers page is looking for a visionary Machine Learning Engineer to join their founding team in San Francisco. The ideal candidate will bridge high-level AI research and real-world physical actuation for autonomous transport platforms. Responsibilities include developing models for trajectory planning, optimizing systems for performance, and managing complex data engineering tasks.

This position offers a salary range of $273,000 – $345,000 per year, along with comprehensive benefits including medical, dental, vision insurance, 401(k), and more.

Qualifications

• 10+ years of non-internship professional MLE experience.
• Deep expertise in applying AI Transformers to robotics.
• Proven track record designing multimodal systems.

Responsibilities

• Research and develop cutting edge RL and distillation techniques for trajectory planning.
• Integrate emerging research to prototype promising solutions.
• Own the post-training lifecycle and optimize models.

Skills

Machine Learning Engineering

AI Transformers

Sensor Fusion

Python

C++

Tools

PyTorch

JAX

Job description

ATOMS Careers page is looking for a visionary Machine Learning Engineer to join their founding team in San Francisco. The ideal candidate will bridge high-level AI research and real-world physical actuation for autonomous transport platforms. Responsibilities include developing models for trajectory planning, optimizing systems for performance, and managing complex data engineering tasks.

This position offers a salary range of $273,000 – $345,000 per year, along with comprehensive benefits including medical, dental, vision insurance, 401(k), and more.