États-Unis
Skyline Scientific Inc
NVIDIA
Senior Manager, Software Engineering - JAX
États-Unis
Santa Clara, California, États-Unis
Traduction automatique. Consultez l’original.
À propos du poste
JAX est le framework de choix pour la recherche IA la plus ambitieuse, et les datacenters sur lesquels il s'exécute évoluent rapidement. Les supercalculateurs NVIDIA deviennent de plus en plus hétérogènes, combinant GPUs, CPUs et LPUs, et JAX doit évoluer pour fonctionner efficacement sur tous. NVIDIA recherche un Senior Engineering Manager pour définir et piloter la stratégie JAX de NVIDIA, en coordonnant plusieurs équipes et flux de travail afin de garantir que JAX offre des performances optimales sur la plateforme, des charges de travail à accélérateur unique à des centaines de milliers d'appareils sur les plus grands supercalculateurs jamais construits.
Une partie essentielle de ce rôle est de s'assurer que JAX est prêt sur les architectures de nouvelle génération, y compris Rubin Ultra, les LPUs et les CPUs qui les pilotent, dès le premier jour. Pour y parvenir, il faut suivre l'évolution de l'IA. Cela signifie soutenir les besoins émergents en matière d'entraînement, post-entraînement, inférence et robotique, en reliant les nouvelles capacités matérielles aux tendances de l'IA. Rejoignez-nous pour construire l'équipe et la technologie derrière les prochaines avancées en IA.
Vos missions
- Piloter la stratégie de contribution d'ingénierie à travers la pile de l'écosystème JAX — y compris le cœur de JAX, XLA, Shardy, MPMD, et les bases de code au niveau applicatif comme MaxText — en priorisant où les contributions de NVIDIA créent un impact maximal sur les performances, l'adoption et l'influence de l'écosystème.
- Promouvoir le travail d'équipe au-delà des frontières organisationnelles pour façonner la plateforme IA de NVIDIA à travers les équipes de compilateurs (XLA, TileIR, CUDA), runtime, réseau et matériel.
- Établir des partenariats solides avec des projets open-source clés, en particulier l'équipe JAX de Google — soutenir nos partenaires tout en représentant les intérêts de NVIDIA dans l'orientation du projet.
- Concevoir des processus et définir des critères de succès clairs pour maintenir l'alignement des équipes sur les résultats.
- Élaborer des plans de staffing, examiner la capacité des équipes et prendre des décisions stratégiques d'embauche pour anticiper les besoins actuels et futurs.
- Diriger, développer et encadrer une organisation d'ingénierie performante — développer des leaders techniques et créer un espace pour que votre équipe expérimente, prototyper et construise des systèmes innovants qui améliorent les produits et les processus.
Ce que nous recherchons
- Master ou Doctorat en Informatique, Génie Informatique ou domaine connexe (ou expérience équivalente).
- 10+ années d'expérience globale en ingénierie logicielle, dont 4+ années en management d'ingénierie ou leadership technique.
- Solide expérience technique en systèmes logiciels, compilateurs, calcul haute performance ou frameworks IA.
- Expérience avérée dans la direction de stratégies et de projets de niveau domaine qui transcendent les frontières organisationnelles, la création de consensus, l'influence sans autorité directe et l'adaptation de la communication à l'audience, des ingénieurs aux cadres.
- Historique de prise en charge des résultats sur plusieurs projets et équipes, d'équilibrage des compromis et de progression avec des informations incomplètes.
- Expérience dans l'identification des lacunes du cycle de développement et la définition de flux de travail entre les équipes et les organisations pour améliorer l'efficacité.
- Expérience dans la direction d'équipes contribuant à des projets à grande échelle avec des collaborateurs externes.
Les atouts qui feront la différence
- Expérience pratique avec JAX et ses systèmes tels que XLA, HLO, PJRT, Shardy, ou le système de traçage et de transformation de JAX.
- Expérience dans la direction d'équipes travaillant sur l'infrastructure de compilateur ou de runtime pour les frameworks de deep learning (JAX, PyTorch, TensorFlow ou équivalent).
- Familiarité avec les composants de la plateforme IA de NVIDIA : programmation GPU, CUDA, cuDNN, NCCL, TensorRT, Transformer Engine et cibles de calcul hétérogènes (GPU, CPU, LPU).
- Expérience dans la conduite d'efforts d'optimisation des performances pour l'entraînement IA distribué à grande échelle, le post-entraînement (RLHF, alignement), l'inférence ou les charges de travail robotiques (multi-nœuds, multi-GPU).
- Expérience dans la conduite de la mise en œuvre de frameworks ou de logiciels sur de nouvelles architectures matérielles, en coordination entre les équipes silicium, pilotes, compilateurs et frameworks.
Votre salaire de base sera déterminé en fonction de votre localisation, de votre expérience et de la rémunération des employés occupant des postes similaires. La fourchette de salaire de base est de 272,000 USD - 431,250 USD.
Vous serez également éligible aux actions et aux avantages sociaux nvidia.com/en-us/benefits/.
Les candidatures pour ce poste seront acceptées au moins jusqu'au 7 août 2026.
Cette offre concerne un poste vacant existant.
NVIDIA utilise des outils d'IA dans ses processus de recrutement.
NVIDIA s'engage à favoriser un environnement de travail inclusif et est fier d'être un employeur garantissant l'égalité des chances. Comme nous accordons une grande importance à la diversité de nos employés actuels et futurs, nous ne faisons aucune discrimination (y compris dans nos pratiques d'embauche et de promotion) fondée sur la race, la religion, la couleur, l'origine nationale, le sexe, l'expression de genre, l'orientation sexuelle, l'âge, l'état civil, le statut d'ancien combattant, le statut de handicap ou toute autre caractéristique protégée par la loi.
Plus d’opportunités avec un seul profil
Vous souhaitez accéder à plus de postes sans répéter tout le processus ? Créez votre profil pour être trouvé par les entreprises utilisant Nort.
Créer mon profil