Autonomous Systems ML Intern

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

Job summary

Atoms is hiring an ML Intern to bridge AI research and real-world actuation for autonomous transport platforms in San Francisco. You will work with senior engineers on production-grade models, edge pipelines, and multi-sensor data systems powering our physical machines.

The role targets students pursuing BS/MS/PhD in related fields with experience in PyTorch, Python, and robotics data processing. The internship is onsite in SF for 12 weeks, with Winter 2027 and Summer 2027 cohorts.

Qualifications

• Currently pursuing a BS, MS, or PhD in Computer Science, Robotics, Electrical Engineering, Data Science, or a related technical field.
• Strong foundation in deep learning concepts and experience with PyTorch or JAX.

Responsibilities

• Collaborate with research engineers to prototype, evaluate, and test emerging Machine Learning and Deep Learning models for trajectory planning and autonomous behavior.
• Help design and evaluate multimodal systems that integrate raw multi-sensor data (Cameras, LiDAR, Radar) to improve spatial-temporal perception.
• Assist in developing interactive world models and simulation tools to re-simulate real-world driving logs and analyze vehicle trajectories.
• Profile and optimize inference pipelines to help run complex models under low latency constraints on vehicle edge hardware.
• Work with data engineering workflows to identify, curate, and structure rare, complex edge cases and long-tail scenarios from physical operations.
• Partner with validation and QA teams to run model releases through simulated scenarios to catch performance regressions and track model behavior.

Skills

Python

Deep learning

PyTorch

C++ basics

Computer Vision

Sensor data

Robotics fundamentals

Education

BS/MS/PhD in CS/Robotics/EE/Data Science

Tools

ROS

Job description

Atoms is hiring an ML Intern to bridge AI research and real-world actuation for autonomous transport platforms in San Francisco. You will work with senior engineers on production-grade models, edge pipelines, and multi-sensor data systems powering our physical machines.

The role targets students pursuing BS/MS/PhD in related fields with experience in PyTorch, Python, and robotics data processing. The internship is onsite in SF for 12 weeks, with Winter 2027 and Summer 2027 cohorts.