Stage Quantification d'Un World Model de Type Jepa pour une Inférence Embarquée-Saclay H/F CEA
- Saclay - 91
- Stage
- 6 mois
- Bac +5
- Industrie high-tech • Telecom
Les missions du poste
En tant que stagiaire au CEA, vous aurez l'opportunité de travailler au sein d'un environnement de recherche de renommée mondiale. Nos équipes sont composées d'experts passionnés et dédiés, offrant un cadre propice à l'apprentissage et à la collaboration. Vous aurez accès à des équipements de pointe et à des ressources de recherche de premier ordre pour mener à bien vos missions. Éventuellement, les travaux menés donneront lieu à une publication scientifique.
Contexte :
Les World Models sont des modèles d'IA qui ouvrent la voie à une nouvelle génération de systèmes autonomes. Ils ont la capacité essentielle de prendre en compte l'impact des actions sur l'environnement permettant ainsi de simuler un grand nombre de séquences d'actions et d'états futurs possibles pour prendre les meilleures décisions. Cependant, les modèles générant des environnements 2D/3D très réalistes nécessitent de vastes bases d'apprentissage, sont trop complexes pour être embarqués sur des systèmes temps réel et restent très sensibles aux bruits des capteurs, ce qui limite leur transfert de la simulation au monde réel. Récemment, les World Models de type JEPA permettent de prédire les caractéristiques essentielles des états futurs à partir d'un espace latent compact, sans avoir à reconstruire les pixels bruts. Ces modèles ont démontré des capacités d'apprentissage, une tolérance au bruit et une vitesse d'inférence remarquables, ce qui en fait de sérieux candidats pour l'autonomie robotique.
Qu'attendons-nous de vous ?
- Prendre en main LeWorldModel[1] dans ses phases d'inférence et d'apprentissage
- Identifier les verrous et contraintes pour son embarquabilité
- Valider ses performances sur un ensemble de datasets représentatifs (ex. en utilisant l'environnement stable-WM[2])
- Effectuer son profiling statique et dynamique
- Identifier et analyser les mécanismes de compression, en particulier la quantification, pouvant s'appliquer à ce modèle en s'appuyant sur les techniques de l'état de l'art et celles déjà mises en oeuvre au sein du laboratoire
- Développer un modèle permettant d'estimer l'impact de la quantification de certaines parties du réseau sur les performances applicatives et sur l'embarquabilité du modèle.
- Le candidat pourra également être amené à valider ses résultats par une exécution réelle sur une des cartes GPU embarquée basse consommation disponible au sein du laboratoire (exemple : Nvidia Jetson Orin).
Pour mener à bien ces missions, le stagiaire pourra s'appuyer sur les connaissances de l'équipe concernant les techniques de quantification ainsi que sur le framework de deep learning, développé au LIAE, pour le profiling, la compression et l'export sur cible matérielle (Aidge[3]).
[1] https://le-wm.github.io/
[2] https://github.com/galilai-group/stable-worldmodel
[3] https://eclipse.dev/aidge/
#Cea List
Le profil recherché
Profil :
Fin d'étude ingénieur ou master 2
Compétences : Python et C++ (obligatoire), Connaissances en deep learning (obligatoire) et en développement sur système embarqué (souhaité)
Les avantages
- Télétravail jusqu’à 3 jours par semaine
- 52 jours de congés/RTT
- Possibilité d’aménagement du temps de travail
- Formation personnalisée
- Restauration d’entreprise
- Offre de transport interne et prise en charge Navigo and co,
- Mutuelle d’entreprise avantageuse
- CE (aides vacances, loisirs, frais de garde, scolarité des enfants etc
Les étapes de recrutement
Les étapes de recrutement peuvent varier selon l'offre à laquelle vous postulez.
-
Dépôt de CV via notre site carrière
-
Préqualification téléphonique
-
Entretiens et évaluation avec manager et RH
-
Négociation salariale et contrat de travail
-
Embauche et intégration
-
CEA en images
Publiée le 12/10/2026 - Réf : 2026-42032