Stage en compilation IA dans l'environnement MLIR

Montbonnot / STAGE

REF : 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 ). 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 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.
• Définir les règles de traduction des dialectes standard MLIR (ex. 'arith' et 'scf' ) vers les nouveaux dialectes définis pendant le stage.

• Bon niveau en C++ et CMake
• Intérêt pour la compilation AI et C/C++
• PyTorch et Python bindings

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

A PROPOS DE KALRAY

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.

Pour plus d'informations, visitez notre site web et suivez-nous sur LinkedIn.