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
- 1Recruiter Screenยท 30 min
- 2Technical Interviewยท 60 min
- 3Hiring Manager Interviewยท 45 min
This description was AI-summarized. View original
0 0 0