Stage en compilation IA dans l’environnement MLIR

Stagiaire compilation IA en environnement MLIRUn(e) stagiaire en compilation IA dans l’environnement MLIRRattaché(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énieurVous êtes curieux(se), ambitieux(se), passionné(e) par la high-tech et les applications issues de nouvelles technologiesVous aimez travailler en équipe et prendre des initiativesVous êtes enthousiaste à l’idée de participer à une aventure humaine et travailler dans un environnement agileVous parlez couramment anglaisType de contrat : Convention de stageGratification : Selon niveau d’étudeConvention collective : MétallurgieAvantages : 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. #J-18808-Ljbffr