Machine Learning Engineer Intern - Scenario GenerationMedium • Santa Clara, California, United States
Machine Learning Engineer Intern - Scenario Generation
Medium
- Santa Clara, California, United States
- Santa Clara, California, United States
About
The intern will explore and prototype a generative simulator that can synthesize diverse, realistic driving scenarios conditioned on map, traffic, and ego-vehicle state.
Responsibilities
Evaluate existing tools for fidelity and efficiency to determine the best baseline.
Ingest and process internal, real-world driving logs to fine-tune the model.
Connect the prototype to the data pipeline, allowing it to seamlessly ingest map data and output structured scenarios.
Translate the generated scenarios into the standardized formats required by the company’s existing simulation engine.
Develop metrics to validate the physical realism, diversity, and kinematic feasibility of the generated scenarios.
Use generated scenarios to stress-test the vehicle's on-board modules against rare or safety-critical events.
Collaborate closely with ML researchers, simulation engineers, and on-board teams to define what makes a scenario useful for their specific needs.
Required Skills
Solid Python programming skills
Familiarity with deep learning frameworks (PyTorch preferred)
Basic understanding of generative models (e.g., VAEs, GANs, diffusion models)
Experience working with datasets (loading, preprocessing)
Preferred Skills
Prior exposure to diffusion models
Experience with autonomous driving datasets
Familiarity with 2D/3D coordinate systems, trajectories, and map representations
Experience with experiment tracking and ML tooling
Basic understanding of scenario-based testing for autonomous driving systems
$19 - $65 an hour
Your opportunities joining PlusAI Work, learn and grow in a highly future-oriented, innovative and dynamic field.
Wide range of opportunities for personal and professional development.
Catered free lunch, unlimited snacks and beverages.
Highly competitive salary and benefits package, including 401(k) plan.
#J-18808-Ljbffr
Languages
- English
Notice for Users
This job comes from a TieTalent partner platform. Click "Apply Now" to submit your application directly on their site.