Drimva

Atoms

AI Research Intern (PhD)

San Francisco, CA +1 · Internship · Posted Oct 22, 2026

Location
San Francisco, CA
San Francisco, California, United States
Education
Currently enrolled in a PhD degree program in Computer Science, EECS, Robotics, AI, or a closely related field.

About the job

Conduct AI research and develop models that help autonomous transport platforms perceive, simulate, plan, and act in real-world environments.

What you'll do

  • Integrate and prototype promising AI research; build multimodal models connecting visual perception and behavioral goals to vehicle actuation.
  • Research reinforcement learning and distillation for trajectory planning, and develop interactive world models from multi-sensor logs.
  • Evaluate autonomous-driving models in simulated scenarios, finding regressions and performance bottlenecks and supporting adaptation to new environments.

What they ask for

  • Currently enrolled in a PhD program in Computer Science, EECS, Robotics, AI, or a closely related field.
  • Publications or active submissions to top-tier AI or robotics venues.
  • PhD-level research background in at least one listed area: generative world models, video generation or diffusion; VLA or embodied AI; RL, trajectory optimization or decision-making; or large-scale multimodal Transformers for robotics or spatio-temporal data.
  • Advanced hands-on proficiency with PyTorch or JAX; proficiency in Python and working knowledge of C++.

Nice to have

  • Research or internship experience in autonomous driving, mobile robotics, or spatial AI; sensor fusion across multi-camera setups, LiDAR, and Radar.
  • Familiarity with BEV representations, 3D occupancy forecasting, multi-task learning, spatial data tokenization, or distributed GPU/TPU training.

Pay

$70/hr per hour (hourly pay)

Tools

PyTorch, JAX, Python, C++, LiDAR, Radar, DeepSpeed, Megatron-LM

AI focus

Research and develop perception, world, and action models for autonomous transport, including multimodal modeling, reinforcement learning, trajectory planning, simulation, and adapting systems to new environments.

Model-written summary · the original posting is authoritative

First seen Oct 29, 2026 · Last verified Nov 5, 2026 (UTC)