Ingénieur(e) compilateur IA
Role details
Job location
Tech stack
Job description
En tant qu'Ingénieur(e) Compilateur IA, vous serez chargé(e) de concevoir, développer et optimiser la chaîne de compilation dédiée à l'intelligence artificielle, faisant le lien entre les frameworks de machine learning de haut niveau et les accélérateurs matériels IA basés sur l'architecture RISC-V.
Vous travaillerez en étroite collaboration avec les équipes Logiciel, Hardware et IA afin d'optimiser l'exécution des modèles d'inférence de nouvelle génération, notamment les grands modèles de langage (LLM), en développant des mécanismes avancés de génération de code et d'optimisation du compilateur.
Vos développements auront un impact direct sur les performances, l'efficacité et la montée en charge de l'ensemble de la chaîne d'inférence IA.
À ce titre, vous serez amené(e) à :
-
Concevoir, développer et maintenir les différents composants du compilateur dédiés aux charges de travail d'inférence IA, avec un focus sur le graph lowering, les optimisations et la génération de code, depuis des frameworks tels que PyTorch, TensorFlow ou ONNX jusqu'à leur exécution sur les accélérateurs matériels.
-
Collaborer étroitement avec l'équipe Architecture afin d'exploiter au mieux les capacités du matériel et de maximiser les performances.
-
Contribuer aux écosystèmes open source de compilation tels que LLVM, MLIR, XLA ou IREE.
-
Analyser, profiler, déboguer et optimiser l'exécution des modèles afin d'améliorer le débit, de réduire la latence et d'optimiser l'efficacité énergétique.
-
Participer à la validation des développements sur silicium prototype, simulateurs ou plateformes matérielles., Team building et autre activités telles que laser game, escape game, voile, ski, karting, karaoké et bien plus encore ! Processus de recrutement : Premier entretien avec le manager Deuxième entretien avec l'équipe Entretien final avec la DRH ou le CEO KALRAY s'engage à promouvoir un environnement de travail fondé sur la diversité et l'inclusion. Nous accueillons les candidatures de personnes issues de tous horizons, identités et expériences. Nous ne pratiquons aucune discrimination (y compris dans nos processus de recrutement et de promotion) fondée sur la religion, la couleur de peau, l'origine, le genre, l'orientation sexuelle, l'âge, le statut marital, le handicap ou toute autre caractéristique protégée par la loi.
Si vous avez besoin d'un aménagement ou d'une adaptation particulière durant le processus d'entretien et au-delà n'hésitez pas à nous en informer. Nous nous engageons à garantir à toutes et à tous les mêmes chances de mettre en valeur leurs compétences et de réussir au sein de notre organisation.
Requirements
Expérience du développement de compilateurs basés sur MLIR, LLVM, IREE ou XLA.
-
Bonne connaissance des formats et frameworks de modèles IA : ONNX, TensorFlow, PyTorch, Hugging Face.
-
Solide expérience des optimisations bas niveau : SIMD, vectorisation, ordonnancement des instructions, data tiling, etc.
-
Bonne compréhension des architectures de calcul accéléré : CPU, GPU, NPU, TPU.
-
Maîtrise des outils de profiling et d'analyse des performances logicielles.
-
Bonne pratique des outils de développement collaboratif (Git, CI/CD).
Atouts appréciés
-
Expérience des architectures RISC-V ou des accélérateurs dédiés à l'IA.
-
Contributions à des projets open source autour de MLIR ou d'autres infrastructures de compilation.
-
Connaissances des techniques de quantification, de calcul en précision mixte et d'optimisation des graphes de calcul.
-
Expérience des logiciels système, des API runtime ou de l'intégration au niveau des pilotes matériels.
-
Connaissances en calcul hétérogène et en programmation parallèle.
Benefits & conditions
Type de contrat : permanent, temps plein Début de contrat : dès que possible Lieu : Montbonnot (38) Nos avantages : Salaire compétitif & attribution d'actions gratuites Jours de congés supplémentaires (RTT) Télétravail Titres-restaurant (Edenred) Mutuelle familiale (Malakoff Humanis) Forfait mobilité durable Avantages congé paternité