Staff Software Engineer - GenAI Performance and Kernel
Databricks
Ce poste de niveau senior consiste à concevoir, implémenter et optimiser des noyaux GPU haute performance pour la pile d'inférence d'IA générative. Le rôle implique le développement de chemins de calcul bas niveau hautement optimisés, la gestion des compromis entre efficacité matérielle et généralité, ainsi que le mentorat d'autres ingénieurs en performance au niveau des noyaux. Le candidat travaillera en étroite collaboration avec des chercheurs en ML, des ingénieurs systèmes et des équipes produits pour améliorer les performances d'inférence à grande échelle. Les responsabilités incluent l'optimisation des noyaux de calcul (attention, MLP, softmax, etc.), l'intégration des optimisations avec les systèmes ML de niveau supérieur, et l'analyse des goulots d'étranglement de performance.
Conditions d'éligibilité
Diplôme BS/MS/PhD en informatique ou domaine connexe
Expérience approfondie dans l'écriture et l'optimisation de noyaux de calcul (CUDA, Triton, OpenCL, LLVM IR, assembly) pour les charges de travail ML
Connaissance solide de l'architecture GPU/accélérateurs (structure warp, hiérarchie mémoire, tensor cores, etc.)
Expérience avec les techniques d'optimisation avancées (tiling, vectorisation, fusion, transformations de boucles, auto-tuning)
Familiarité avec les bibliothèques de noyaux ML (cuBLAS, cuDNN, CUTLASS, etc.)
Compétences en débogage et profilage (Nsight, NVProf, perf, vtune)
Expérience dans l'intégration de noyaux optimisés dans des systèmes d'inférence ML réels
Excellentes compétences en communication et leadership
Pièces à fournir
- CV
- Lettre de motivation
Candidatez à cette offre avec un dossier généré par l'IA
Matchwork rédige votre CV et votre lettre de motivation personnalisés pour cette offre en quelques minutes.
Créer mon dossier gratuitement