Poste Ingénieur Compilation pour le Machine Learning H/F INRIA

Poste Ingénieur Compilation pour le Machine Learning

Type de contrat : CDD

Contrat renouvelable : Oui

Niveau de diplôme exigé : Bac +5 ou équivalent

Autre diplôme apprécié : Thèse en informatique ou mathématiques appliquées

Fonction : Ingénieur scientifique contractuel

Niveau d'expérience souhaité : De 3 à 5 ans

A propos du centre ou de la direction fonctionnelle

Le Centre Inria de l'Université Grenoble Alpes, regroupe un peu moins de 600 personnes réparties au sein de 26 équipes de recherche et 9 services support à la recherche.

Son effectif est distribué sur 3 campus à Grenoble, en lien étroit avec les laboratoires et les établissements de recherche et d'enseignement supérieur (Université Grenoble Alpes, CNRS, CEA, INRAE...), mais aussi avec les acteurs économiques du territoire.

Présent dans les domaines du calcul et grands systèmes distribués, logiciels sûrs et systèmes embarqués, la modélisation de l'environnement à différentes échelles etla science des données et intelligence artificielle, le Centre Inria de l'Université Grenoble Alpes participe au meilleur niveau à la vie scientifique internationale par les résultats obtenus et les collaborations tant en Europe que dans le reste du monde.

Contexte et atouts du poste

Ingénieur de recherche et et dévelopement en compilation pour un framework de code génération pour le Machine Learning.

Travail au sein de l'équipe de recherche INRIA/CORSE sur l'outil XTC (Xdsl Transform Compiler) dans le cadre du project DeepGreen (2024-2027).

Thèmes de recherche : compilation, architecture, langages domaine spécifique, machine learning

Mission confiée

Mission

Contribution à l'amélioration des performances (temps d'inférence, taille de code) de la chaine d'outils XTC sur des graphes de type CNN/DNN/LLM.

Localisation

Travail avec l'équipe INRIA/CORSE sur Grenoble Antenne Minatec.

Collaborations

Collaboration avec les collaborateurs académiques de l'équipe et les partenaires du projet DeepGreen.

Principales activités

Principales Activités :
- Extension du langage d'optimisation de XTC :
- Définition de nouvelles directives de transformation, du type :
- Fusion d'opérateurs,
- Packing de tenseurs,
- Pipeline logiciel,
- Support de dimensions de tenseurs paramétriques
- Implémentation dans les différents back-ends : MLIR et TVM
- Amélioration des algorithmes de recherche de solutions dans l'auto-tuner de XTC :
- Calcul de métriques de performance à partir des transformations, du type :
- Estimation d'intensité opérationelle
- Estimation de paralllelisme niveau tâche et niveau block de base
- Utilisation de modèles de prédiction statistiques basé sur de métriques de performance
- Intégration de contraintes sur l'espace de recherche

Autres Activités :
- Amélioration des parties basses du compilateur pour supporter les transformations et la génération de code XTC :
- Implementations si nécessaire de passes dans MLIR/TVM et/ou LLVM
- Amélioration des performances de l'emission vers C (MLIR vers C et TVM vers C)
- Veille Technologique compilation pour le ML

Compétences

Compétences requises :
- Compilation, optimisation de performance sur CPU/GPU
- Architecture logicielle, architecture des ordinateurs
- Representations intermédiaires de compilateur, i.e. MLIR/LLVM-IR/TVM-IR
- C/C++/python/assembleur ARM

Compétences souhaitées :
- Analyse/transformations de graphes de calculs
- Connaissance des plateformes de dévelopement pour l'IA
- Dévelopement pour plateformes embarquées

Avantages
- Restauration subventionnée
- Transports publics remboursés partiellement
- Congés : 7 semaines de congés annuels + 10 jours de RTT (base temps plein) + possibilité d'autorisations d'absence exceptionnelle (ex : enfants malades, déménagement)
- Possibilité de télétravail (90 jours par an flottants) et aménagement du temps de travail
- Prestations sociales, culturelles et sportives (Association de gestion des oeuvres sociales d'Inria)
- Accès à la formation professionnelle
- Participation employeur mutuelle santé (sous conditions)

Rémunération

A partir de 2 692 € brut mensuel selon expérience et diplômes.