Ingénieur d'infrastructure ML

À propos de Nébius :

Nebius ouvre une nouvelle ère en matière d'infrastructure cloud pour l'économie mondiale de l'IA. Nous construisons une plate-forme cloud d'IA complète qui prend en charge les développeurs et les entreprises, depuis la formation des données et des modèles jusqu'au déploiement en production, sans le coût et la complexité liés à la création d'une grande infrastructure interne d'IA/ML.

Construit par des ingénieurs, pour des ingénieurs. De l’orchestration GPU à grande échelle à l’optimisation des inférences, nous maîtrisons les problèmes difficiles liés au calcul, au stockage, à la mise en réseau et à l’IA appliquée.

Coté au Nasdaq (NBIS) et basé à Amsterdam, nous avons une présence mondiale avec des pôles de R&D en Europe, au Royaume-Uni, en Amérique du Nord et en Israël. Notre équipe de plus de 1 500 personnes comprend des centaines d’ingénieurs possédant une expertise approfondie en matière de R&D sur le matériel, les logiciels et l’IA.

Le rôle

Nous recherchons un ingénieur ML/IA hautement qualifié pour rejoindre notre équipe afin de diriger et de prendre en charge l'analyse comparative des plates-formes GPU. L'analyse comparative des plates-formes GPU pour les charges de travail d'apprentissage automatique et d'IA. Vous jouerez un rôle essentiel dans l'évaluation des performances du matériel basé sur GPU pour divers frameworks d'apprentissage en profondeur et d'IA, permettant des décisions basées sur les données pour l'optimisation de la plate-forme et le développement de matériel de nouvelle génération.

Vos responsabilités comprendront :

• Travailler en étroite collaboration avec les équipes de développement et de matériel pour profiler et analyser les performances du GPU au niveau du système et du noyau.

• Évaluer et comparer les performances du GPU sur différentes plates-formes, architectures et piles logicielles (par exemple, CUDA, ROCm).

• Déboguez et optimisez les charges de travail ML pour qu'elles s'exécutent efficacement sur le matériel GPU, en identifiant et en résolvant les goulots d'étranglement en matière de performances.

• Effectuer des tests d'acceptation pour les nouveaux clusters GPU, en garantissant que le matériel et les logiciels répondent aux exigences de performances, de stabilité et de compatibilité pour les charges de travail d'IA.

• Réalisez des expériences sur diverses configurations de systèmes GPU pour évaluer l'impact des différentes stratégies d'interconnexion et des optimisations au niveau du système sur les performances et l'évolutivité.

• Développer des outils et des tableaux de bord pour visualiser les mesures de performances, les goulots d'étranglement et les tendances.

• Contribuer aux outils, aux cadres et aux meilleures pratiques internes.

Nous attendons de vous que vous ayez :

• Une compréhension approfondie des fondements théoriques de l'apprentissage automatique

• Compréhension approfondie des aspects de performance de la formation et de l'inférence de grands réseaux de neurones (parallélisme données/tenseur/contexte/expert, déchargement, noyaux personnalisés, fonctionnalités matérielles, optimisations de l'attention, traitement par lots dynamique, etc.)

• Expérience approfondie des frameworks d'apprentissage profond modernes (PyTorch, JAX, Megatron-LM, Tensort-LLM)

• Bonne compréhension de la pile GPU : CUDA, NCCL, pilotes et bibliothèques pertinentes

• Familiarité avec les environnements conteneurisés (par exemple, Docker, Kubernetes).

• Forte communication et capacité à travailler de manière indépendante

Façons de se démarquer de la foule :

• Familiarité avec les frameworks d'inférence LLM modernes (vLLM, SGLang, TensorRT)

• Expérience avec Python et les outils de profilage de performances (par exemple, Nsight, nvprof, perf).

• Familiarité avec les plateformes cloud ML telles qu'AWS, GCP, Azure ML

• Contributions aux outils d'analyse comparative du ML open source

Avantages et avantages :

• Rémunération compétitive

• Possibilités d'évolution de carrière et d'apprentissage

• Flexibilité et appropriation

• Culture collaborative et innovante

• Possibilité de travailler sur des projets d'IA percutants

• Environnement international et équipes talentueuses

Qu'est-ce que ça fait de travailler chez Nebius :

Evolution rapide - Réflexion audacieuse - Croissance constante - Impact significatif - Confiance et appropriation réelle - Opportunité de façonner l'avenir de l'IA

Déclaration sur l'égalité des chances :

Nebius est un employeur garantissant l'égalité des chances. Nous nous engageons à favoriser un lieu de travail inclusif et diversifié et à offrir des opportunités d'emploi égales dans tous les aspects de l'emploi. Nous ne faisons aucune discrimination fondée sur la race, la couleur, la religion, le sexe (y compris la grossesse), l'origine nationale, l'ascendance, l'âge, le handicap, les informations génétiques, l'état civil, le statut d'ancien combattant, l'orientation sexuelle, l'identité ou l'expression de genre, ou toute autre caractéristique protégée par la loi applicable.

Les candidats doivent être autorisés à travailler dans le pays dans lequel ils postulent et devront fournir une preuve d'éligibilité à l'emploi comme condition d'embauche.

Si vous avez besoin d'aménagements pendant le processus de candidature, veuillez nous en informer.

Source : Jobicy (https://jobicy.com/jobs/155964-ml-infrastructure-engineer)