Stage en compilation IA dans l’environnement MLIR

Kalray; Montbonnot-Saint-Martin; Hybride; EUR 40 000 - 60 000; Temps partiel

Résumé du poste

Une entreprise technologique européenne recherche un(e) stagiaire en compilation IA dans l'environnement MLIR, basé à Montbonnot. Le stage permettra d'explorer la génération de code machine à partir de MLIR, avec des missions axées sur l'architecture 'manycore' et les défis de l'intelligence artificielle. Un environnement de travail hybride est proposé, ainsi que des avantages attractifs comme des activités de teambuilding.

Prestations

Travail hybride

Titres restaurant

Teambuilding et autres activités

Qualifications

• Curiosité et ambition dans le domaine des nouvelles technologies.
• Travail en équipe et initiative dans un environnement agile.
• Anglais courant requis pour le poste.

Responsabilités

• Immerger dans l'environnement MLIR et LLVM.
• Définir de nouveaux dialectes MLIR pour décrire les caractéristiques des CPU.

Connaissances

Curiosité

Esprit d'équipe

Passion pour la high-tech

Anglais courant

Formation

Étudiant en 3ème année d'école d'ingénieur

Description du poste

Stagiaire compilation IA en environnement MLIR

Un(e) stagiaire en compilation IA dans l’environnement MLIR

Rattaché(e) au Directeur de l’Innovation, vous participez à une chaine de compilation AI partant de PyTorch et basée sur l’environnement MLIR (https://www.stephendiehl.com/posts/mlir_introduction/ ). Le sujet à explorer concerne la génération de code machine directement depuis MLIR, sans repasser par LLVM ou GCC. Cette approche a été montrée faisable et utile dans le cas de cœurs RISC-V augmentés d’opérateurs pour accélérer l’inférence de modèles IA, voir “A Multi-Level Compiler Backend for Accelerated Micro-Kernels Targeting RISC-V ISA Extensions” by Lopoukhine et al. (https://arxiv.org/abs/2502.04063 ). Il s’agit d’une opportunité de découvrir ou d’enrichir vos connaissances en environnement de compilation pour l’IA et de mettre en pratique la théorie.

A ce titre, vos missions principales sont les suivantes :

• S’immerger dans l’environnement MLIR et LLVM (build system, classes de containers, TableGen)
• Définir le ou les nouveaux dialectes MLIR qui permettent de décrire les caractéristiques d’un CPU tels que les contraintes d’allocation des registres et celles d’ordonnancement des instructions.
• Stagiaire en 3ieme année d’école d’ingénieur
• Vous êtes curieux(se), ambitieux(se), passionné(e) par la high-tech et les applications issues de nouvelles technologies
• Vous aimez travailler en équipe et prendre des initiatives
• Vous êtes enthousiaste à l’idée de participer à une aventure humaine et travailler dans un environnement agile
• Vous parlez couramment anglais
• Type de contrat : Convention de stage
• Gratification : Selon niveau d’étude
• Convention collective : Métallurgie
• Avantages : Travail hybride, titres restaurant, teambuilding et autre activités telles que laser game, escape game, voile, ski, karting, karaoké et bien plus encore !
• Lieu : Montbonnot (38)

Essaimage du CEA fondé en 2008, Kalray est un acteur européen de référence dans le domaine des semi-conducteurs, soutenu par des investisseurs industriels de premier plan tels que Safran, NXP Semiconductors et Bpifrance. Avec plus de 30 familles de brevets et plus de 100 millions d’euros investis en R&D, notre technologie ouvre la voie à une nouvelle ère dans le traitement des données.

Grâce à leur architecture « manycore » brevetée, les processeurs intelligents MPPA® de Kalray (dits DPU ou xPU) sont capables de gérer en parallèle – et sans goulets d’étranglement – de multiples flux de données, permettant ainsi aux applications qui exploitent des quantités massives de données d’être plus intelligentes, plus efficaces et économes en énergie, en complément des approches traditionnelles (CPU et GPU).

L’offre de Kalray comprend des solutions d’accélération matérielle et logicielle, ainsi qu’une offre de co-développement – ou de design à la demande – de processeurs et de solutions d’accélération optimisées pour des usages spécifiques dans des secteurs à forte croissance tels que les data centers, les gigafactories d’IA, les réseaux 5G, l’industrie 4.0, le spatial et la défense, et bien d’autres.

Kalray est avant tout une équipe de personnes passionnées, enthousiastes et volontaires, qui placent l’excellence et la collaboration au cœur de leur travail. Que vous soyez en stage ou en alternance, nous mettrons tout en œuvre pour vous aider à exprimer pleinement votre talent, développer votre potentiel et contribuer au succès de l’entreprise.

Nos bureaux sont situés près de Grenoble, dans les Alpes françaises, et nous disposons de bureaux supplémentaires près de Nice, à Sophia Antipolis, ainsi qu’en Roumanie.