Internship in Machine Learning (m/f) - 50-100%

Implementing & Evaluating new ML Models for Geometry Generation

In this internship, you will have the opportunity to review, implement, and evaluate the latest machine learning models in the realm of geometry generation for process engineering and manufacturing. The nature of this position is inherently experimental and exploratory: you will dive into state-of-the-art neural network architectures, adapt them to our unique context in process engineering, and assess their performance against the real-world needs of our customers.

You will collaborate closely with a dedicated product team while also receiving guidance from ML engineers through regular meetings and knowledge-sharing sessions. In addition to researching and refining advanced ML approaches, you will occasionally contribute to the broader ML product lifecycle. Should your models prove successful, they may be integrated into our final product or prototype, offering a tangible impact on the cutting-edge solutions we deliver.

Your tasks

• Review the state-of-the-art methods and models applicable to our domain and present them to the team
• Implement these models in a prototypical manner in Python and PyTorch, train them on our own custom datasets, and evaluate them. Adapt them to our needs with hyperparameter tuning or architectural modifications

• Support the product team in various ways by contributing to the end-to-end ML lifecycle (e.g. data generation, pre-processing, creating experiment pipelines, etc.)Research suitable tools and frameworks for autonomous coding

Your profile

• Master’s degree (or in the process of obtaining one) in computer science, data science, or engineering
• Very good programming skills in Python and familiarity with at least one of the common machine learning frameworks like PyTorch, TensorFlow, or JAX and C++ would be a plus
• Having at least one ML project under your belt that goes beyond the scope of small lecture projects
• Knowledge of modern neural architecture, especially in computer vision & graphics and 3D geometry would be a plus
• Strong technical interest, willingness to get involved in complex technical issues
• Team Player with a strong commitment to problem solving

• Good knowledge of English required. German would be a plus

Our offer

• A challenging role in a respectful and global working environment
• Being part of a world-leading, high-end CAE software company
• Working on cutting edge technologies for the automotive industry
• Professional development and continuous education

• Excellent balance between your working life and your private life through flexible working hours

This role is based in Zurich. The duration at 100% workload is 6 months. The minimum workload is 50%, extending the duration accordingly.

Start of work is as soon as possible. Please note that we only consider applicants that have the legal right to work in the respective location. We will only consider direct applications. Please send us your CV and attachments. We are looking forward to receiving your application.