ML/AI Research Intern: Real-World Autonomous Systems (Onsite)

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

Job summary

Atoms is building the machines that power the next era of progress. We seek a visionary AI/ML Research Intern to join our founding team and bridge high-level AI research with real-world physical actuation for autonomous transport platforms.

You will contribute to World Models and Vision-Language-Action projects, translating breakthroughs into scalable deployments that help our platforms perceive, simulate, and navigate complex environments.

Qualifications

• PhD student in CS/EECS/Robotics/ML or closely related field.
• Publications or submissions in top ML/robotics venues.

Responsibilities

• Research and develop RL and distillation for trajectory planning.
• Prototype multimodal models translating perception to vehicle actuation.
• Develop world models from multi-sensor data for re-simulation.
• Ensure models adapt to urban environments and edge cases.
• Run experiments in simulated scenarios and identify bottlenecks.
• Collaborate with validation and research teams on deployments.

Skills

VLA models

RL

Transformers

PyTorch

JAX

Python

C++

Education

PhD in Computer Science / EECS / Robotics / ML

Tools

DeepSpeed

Megatron-LM

Ray

PyTorch FSDP

Job description

Atoms is building the machines that power the next era of progress. We seek a visionary AI/ML Research Intern to join our founding team and bridge high-level AI research with real-world physical actuation for autonomous transport platforms.

You will contribute to World Models and Vision-Language-Action projects, translating breakthroughs into scalable deployments that help our platforms perceive, simulate, and navigate complex environments.