Autonomous Systems ML Intern: Edge AI & Perception

Atoms; 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. As an ML Intern in San Francisco, you’ll bridge research and real-world actuation for autonomous transport platforms, collaborating with senior engineers to contribute production-grade models and real-time data systems.

You will work on multimodal sensor data, edge inference, and world models, gaining hands-on experience across hardware and software stacks while improving safety and efficiency of automated systems.

Qualifications

• Pursuing BS, MS, or PhD in Computer Science, Robotics, Electrical Engineering, Data Science, or related field.
• Strong foundation in deep learning concepts and experience with PyTorch or JAX.
• Hands-on Python programming; C++ a plus.

Responsibilities

• Prototype, evaluate, and test ML/DL models for trajectory planning and autonomous behavior.
• Design and evaluate multimodal systems with sensors (Cameras, LiDAR, Radar).
• Develop world models and simulation tools to re-simulate driving logs and trajectories.

Skills

Python

Deep learning

PyTorch

JAX

C++

Education

CS/Robotics/EE/DS degree

Job description

Atoms is building the machines that power the next era of progress. As an ML Intern in San Francisco, you’ll bridge research and real-world actuation for autonomous transport platforms, collaborating with senior engineers to contribute production-grade models and real-time data systems.

You will work on multimodal sensor data, edge inference, and world models, gaining hands-on experience across hardware and software stacks while improving safety and efficiency of automated systems.