Autonomous Driving AI Research Intern

ATOMS Careers page; Mountain View (CA); On-site; USD 83,000 - 124,000; Part time

Job summary

Atoms is building the next era of Physical AI with real-world robots for food, mining, and transport. We’re seeking an AI Research Intern to bridge high-level AI research with actuation for autonomous transport platforms.

You will contribute to core initiatives in perception, world models, and action models, translating breakthroughs into scalable deployment on real systems. You’ll adapt and evaluate RL and planning methods, train multimodal models, and help improve simulation fidelity for

Qualifications

• Currently enrolled in a Bachelor's or Master's degree program in Computer Science, EECS, Robotics, AI, or a closely related field.
• Coursework, project experience, or research background in Vision-Language-Action (VLA) models or embodied AI.
• Proficiency in Python and working knowledge of C++.

Responsibilities

• Integrate and adapt emerging AI research into internal model pipelines, validating performance on real-world datasets and simulation.
• Prototype and evaluate RL, trajectory planning, and distillation algorithms based on state-of-the-art literature.
• Train and fine-tune multimodal models that map real-time sensor inputs and behavioral goals to vehicle actuation.
• Experiment with world model architectures using multi-sensor log data to improve simulation fidelity and trajectory querying.
• Analyze edge cases and failure modes in novel driving environments to iterate on model performance and stability.
• Run rigorous simulation experiments alongside research and validation teams to identify performance bottlenecks and track model improvements.

Skills

Vision-Language-Action models

Reinforcement Learning

Multimodal Transformers

PyTorch

JAX

Python

C++

Education

Bachelor's or Master's in CS / EECS / Robotics / AI

Tools

PyTorch

JAX

Job description

Atoms is building the next era of Physical AI with real-world robots for food, mining, and transport. We’re seeking an AI Research Intern to bridge high-level AI research with actuation for autonomous transport platforms.

You will contribute to core initiatives in perception, world models, and action models, translating breakthroughs into scalable deployment on real systems. You’ll adapt and evaluate RL and planning methods, train multimodal models, and help improve simulation fidelity for