Informations générales
Entité de rattachement
La Direction des Applications Militaires (DAM) du CEA, au cœur des enjeux de la dissuasion nucléaire Française, cherche ses futurs talents. Organisme inclusif, le CEA est handi-accueillant : nos emplois sont ouverts à toutes et tous.
Associer les forces et les compétences de chacun pour atteindre nos objectifs est l'une de nos valeurs partagées par nos 4 600 salariés, répartis sur 5 centres.
Les 1 000 salariés du centre du Cesta, en Nouvelle-Aquitaine, participent au développement des armes de la force de dissuasion française.
Le centre assure le rôle d'architecte industriel des têtes nucléaires mises à la disposition des Armées françaises depuis leur conception jusqu'à leur retrait du service.
Pour garantir les performances opérationnelles des systèmes d'armes, le Cesta s'inscrit dans une démarche de simulation de haut niveau et s'appuie sur un parc de moyens d'essais exceptionnels. Parmi ceux-ci, il développe et exploite le plus grand laser d'Europe : le Laser MégaJoule (LMJ).
Venez vous investir au service de la Défense et de la Sécurité de notre pays, relever des défis scientifiques et techniques avec des moyens technologiques d'exception ! Vous contribuerez à l'excellence de la recherche et à la compétitivité de l'industrie française en construisant votre avenir dans un environnement varié et respectueux.
Référence
2026-41609-S2213
Description du poste
Domaine
Mathématiques, information scientifique, logiciel
Contrat
Stage
Intitulé de l'offre
Stage - Bac+5 - Evaluation d'algorithmes en précision mixte sur GPU pour un solveur dense - H/F
Sujet de stage
L'unité d'accueil située au CEA/CESTA élabore des modèles physico-numériques multi physiques, multi-échelles et développe des codes de calcul dans les domaines de l’aérodynamique hypersonique, de l’électromagnétisme et de la dynamique rapide. Ces développements bénéficient des approches les plus modernes du génie logiciel et sont conduits dans le contexte du calcul haute performance afin de tirer le meilleur parti des supercalculateurs de la DAM.
Lors de ce stage, vous serez amené(e) à travailler sur un solveur linéaire dense à coefficients complexes pour résoudre des problèmes d'électromagnétisme. Du aux besoins de l'IA, le nombre d'unités de calcul en faible précision sur les nouveaux modèles de GPU ne cesse d'augmenter d'une génération à l'autre. Il devient donc crucial d'exploiter ces unités pour améliorer les performances du solveur sur ces architectures, tout en maintenant une haute précision sur les résultats, critique pour les applications scientifiques
Durée du contrat (en mois)
5-6 mois
Description de l'offre
Des travaux précédents ont montré que l'introduction d'algorithmes en précision mixte dans le solveur permettent d'améliorer les performances tout en maintenant une précision suffisante. Cependant, les bibliothèques d'algèbre linéaire optimisées fournies par les vendeurs (e.g. cuBLAS/rocBLAS) n'offrent qu'un support très limité pour la précision mixte en arithmétique complexe.
Une première stratégie pour résoudre ce problème consiste en l'implémentation de noyaux en précision mixte complexe optimisés à l'aide de librairies telles que CUTLASS. La seconde stratégie envisagée consiste en la réécriture des algorithmes complexes en une succession de noyaux réels (approche 3M), qui nous permettrait donc de bénéficier des routines en précision mixte des bibliothèques des vendeurs. Le(la) candidat(e) aura pour mission d'implémenter ces stratégies dans le solveur afin d'évaluer leurs performances et leur précision.
En fonction du temps et des préférences du(de la) candidat(e), on pourra également explorer l'utilisation de méthodes d'émulation flottante de type BF16x9 ou Ozaki qui permettent d'émuler des hautes précisions à partir de calculs en faibles précisions.
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
Calcul Haute Performance, Parallélisme, Programmation GPU
Bac+5
Localisation du poste
Site
Cesta
Localisation du poste
France, Nouvelle-Aquitaine
Create candidate area
