Atoms

Machine Learning Engineer Intern

Atoms
Apply
5 hours ago

Responsibilities

  • Prototype, evaluate, and test machine learning and deep learning models for trajectory planning and autonomous behavior.
  • Design and evaluate multimodal systems integrating camera, LiDAR, and radar data for spatial-temporal perception.
  • Develop interactive world models and simulation tools to replay driving logs and analyze vehicle trajectories.
  • Profile and optimize inference pipelines for low-latency execution on vehicle edge hardware.
  • Identify, curate, and structure rare edge cases and long-tail scenarios using data engineering workflows.
  • Partner with validation and QA teams to test model releases in simulated scenarios, detect regressions, and track model behavior.

Requirements

  • 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 and experience with PyTorch or JAX through coursework, research, or prior internships.
  • Hands-on programming experience in Python; familiarity with C++ is a plus.
  • Coursework, project experience, or research in computer vision, spatial-temporal modeling, reinforcement learning, model optimization, model evaluation, or data engineering.
  • Familiarity with robotics data structures or multisensor processing involving cameras, LiDAR, or radar is highly preferred.
  • Interest in solving complex real-world engineering problems and deploying AI into physical systems.

Benefits

  • 12-week internship program with Winter 2027 and Summer 2027 cohorts.
  • Role is based in San Francisco and requires onsite work five days per week.
  • Winter 2027 cohort runs January through April; Summer 2027 cohort runs late May through September.

Categories

Atoms

About Atoms

501-1,000 employees

Atoms develops robotics, software, and infrastructure to automate real-world industrial operations, with offerings focused on food production, mining, and robotic transport platforms. It sells integrated automation systems and platforms—Atoms Food, Atoms Mining, and a transport wheelbase for robots—to industrial customers seeking greater productivity and scalability. The privately held company is headquartered in Los Angeles.

Contact me