Job Description
High-fidelity traffic scenario generation is increasingly important for developing and validating autonomous driving systems. Diffusion models have recently demonstrated strong potential for synthesizing diverse and realistic multi-agent driving scenes.
In this internship, you will contribute to advancing an existing diffusion-based traffic scene generation model used in closed-loop simulation within the Aumovio AI Lab in Berlin. The focus of your work will be on improving the model's generation quality through enhanced training strategies, better data representations, and the development of a critic module for automated quality assessment.
You will have direct access to our simulation stack, enabling interactive testing and evaluation of your improved models across a wide range of simulated driving scenarios.
Duration: 6 Months
Tasks:
* Conduct a familiarization phase and baseline analysis of the current diffusion model.
* Enhance the existing diffusion model using improved conditioning, data augmentation, fine-tuning, or reinforcement learning methods.
* Develop a critic model capable of assessing scene realism, safety, and consistency.
* Integrate the improved generator and critic into the simulation framework and conduct a comprehensive evaluation in closed-loop settings.
Qualifications
* Enrolled students (m/f/diverse) in Computer Science, Electrical Engineering, Robotics, AI, or a related field.
* Solid programming skills in Python and experience with deep learning frameworks (preferably PyTorch).
* Prior experience with diffusion models or generative modeling is a strong advantage.
* Strong command of English (spoken and written); German is a plus.
* Independent, reliable, and collaborative working style with strong problem-solving skills.
Please remember to upload your current certificate of enrollment as well as your current transcript of grades and an excerpt from the current study regulations regarding completion of a thesis, as these are mandatory for processing your application. If required, please submit your valid residence permit as well as your work permit including the additional sheet.
Applications from severely handicapped people are welcome.
Additional Information
* Flexible working hours / working time models (job flexibility)
* A performance-based remuneration for your work with us
* Qualified support by our experts
* Health-oriented workplace
* Health courses and sports activities (EGYM Wellpass)
* Subsidy Germany Ticket
* Regular interns table, if applicable
* Regular (team) events, if applicable
* Company restaurant, if applicable
Diversity, Inclusion & Belonging are important to us and make our company strong and successful. We offer equal opportunities to everyone -regardless of age, gender, nationality, cultural background, disability, religion, ideology or sexual orientation.
Ready to take your career to the next level? The future of mobility isn’t just anyone’s job. Make it yours! Join AUMOVIO. Own What’s Next.
Ready to take your career to the next level? The future of mobility isn’t just anyone’s job. Make it yours! Join AUMOVIO. Own What’s Next.