Stage RL et Graph-JEPA pour lissage de maillages 2D/3D

CEA; Bruyères-le-Châtel; On-site; EUR 17,000 - 21,000; Full time

Job summary

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

Qualifications

• É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.

Responsibilities

• 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.

Skills

Python

RL

Deep Learning

Education

Bac+5

Tools

PyTorch

Job description

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