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