Poste ingénieur compilation pour le machine learning

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

1. Restauration subventionnée
2. Transports publics remboursés partiellement
3. 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)
4. Possibilité de télétravail (90 jours par an flottants) et aménagement du temps de travail
5. Prestations sociales, culturelles et sportives (Association de gestion des œuvres sociales d'Inria)
6. Accès à la formation professionnelle
7. Participation employeur mutuelle santé (sous conditions)

Rémunération

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