5d ago

Machine Learning Engineer Intern - Planning

Santa Clara, CA

$39.5k-$135.2k / year

internshipinterntransportation

๐Ÿ›  Tech Stack

๐Ÿ’ผ About This Role

You'll expand scene transformer models to generate multiple ego-vehicle trajectories in ambiguous cases and explore recursive multi-step losses for longer horizon predictions. Your work will directly impact safe autonomous truck deployment.

๐ŸŽฏ What You'll Do

  • Modify scene transformer for multimodal ego trajectory generation
  • Formulate recursive and multi-step loss functions
  • Mitigate mode collapse for diverse feasible paths
  • Design metrics for trajectory diversity and safety

๐Ÿ“‹ Requirements

  • Expertise in trajectory prediction and behavior forecasting
  • Experience training large-scale models with self-supervision
  • Practical experience with PyTorch, Lightning, or JAX
  • Understanding of AV stack components: localization, perception, planning

โœจ Nice to Have

  • Experience with uncertainty estimation (aleatoric, epistemic)
  • Experience with multi-modal prediction and probabilistic modeling

๐Ÿ“จ Hiring Process

Estimated timeline: 2-4 weeks ยท AI estimate

  1. 1Recruiter Screenยท 30 min
  2. 2Technical Interviewยท 60 min
  3. 3Hiring Manager Interviewยท 45 min

This description was AI-summarized. View original

0 0 0