Stage - Bac+5 - Apprentissage par Renforcement et Modèles de Monde pour les maillages - H/F

CEA; Bruyères-le-Châtel; Presencial; EUR 17.000 - 21.000; Jornada completa

Descripción de la vacante

CEA recherche un(e) étudiant(e) Bac+5 pour développer une architecture hybride où un agent d’apprentissage par renforcement déplace les sommets d’un maillage, en s’appuyant sur les prédictions de Graph-JEPA. Le stage vise à apprendre des représentations latentes liées aux opérations de maillage sur cartes 2D/3D et à évaluer la qualité des maillages.

Le candidat doit maîtriser Python et le deep learning (PyTorch), avec une connaissance souhaitée de Graph-JEPA, et sera intégré au site DAM

Formación

• Étudiant(e) Bac+5 recherché pour un stage dans le domaine du maillage et du RL.
• Maîtrise de Python et frameworks ML recommandée.
• Connaissances en graph-JEPA ou en ml sur graphes appréciées.

Responsabilidades

• Développer une architecture hybride où un agent RL déplace les sommets du maillage en s’appuyant sur Graph-JEPA.
• Évaluer l’impact des déplacements sur la qualité des maillages (forme, jacobien, angles).
• Implémenter un agent RL avec une approche acteur-critique et une fonction de récompense corrélée aux métriques de maillage.
• Comparer les résultats avec des algorithmes de lissage existants et analyser la convergence.

Conocimientos

Python

RL

Deep Learning

Educación

Bac+5

Herramientas

PyTorch

Descripción del empleo

En considérant un problème plus simple de maillage, à savoir le lissage qui consiste à déplacer les sommets d’un maillage pour améliorer la qualité géométrique des cellules composant le maillage, l’étudiant(e) aura pour objectif de développer et évaluer une architecture hybride où un agent RL apprend à déplacer les sommets du maillage en s’appuyant sur les prédictions abstraites de Graph-JEPA. L’idée sous-jacente est d’apprendre des représentations latentes relatives à des opérations de maillage sur des cartes combinatoires de dimension 2 et 3.

Les grandes étapes du travail proposé seront :

• Assimiler les concepts suffisants en maillage et comprendre le formalisme des cartes combinatoire.
• Concevoir une architecture JEPA adaptée. Le modèle devra apprendre à prédire l’impact latent d’un déplacement de sommet sans reconstruire le maillage complet.
• Implémenter un agent de RL avec une approche acteur-critique et une fonction de récompense qui sera directement corrélée à des métriques de qualité de mailles (ex : rapport de forme, jacobien des éléments, régularité des angles) en utilisant l’architecture précédemment définie.
• Comparer les résultats obtenus avec des algorithmes de lissage usuels, aussi bien en termes de qualité finale du maillage et de temps de convergence.

Conformément aux engagements pris par le CEA en faveur de l'intégration des personnes handicapées, cet emploi est ouvert à toutes et à tous. Le CEA propose des aménagements et/ou des possibilités d'organisation pour l’inclusion des travailleurs handicapés.

Participant à la protection nationale, une enquête administrative est réalisée pour tous les collaborateurs du CEA afin d'assurer l'intégrité et la sécurité de la nation.

Profil du candidat

Python, Apprentissage par renforcement, apprentissage profond, pytorch

Bac+5

Localisation du poste

Site

DAM Île-de-France

Localisation du poste

France, Ile-de-France, Essonne (91)

Ville

Bruyères-le-Châtel