Software Engineer, PyTorch - Deep Learning

Société : NVIDIA

Poste : Ingénieur Logiciel, PyTorch - Apprentissage Profond

Lieu : Ouvert aux talents du monde entier

Êtes-vous prêt à innover à l'intersection de l'IA et du génie logiciel ? NVIDIA recherche des Ingénieurs en Compilation pour l'Apprentissage Profond compétents pour créer des outils de pointe améliorant les capacités des ingénieurs en IA dans le monde entier. Saisissez l'opportunité d'impacter la conception, le développement et le déploiement de l'IA à l'échelle mondiale, en utilisant des milliers de GPU. Rejoignez notre équipe dédiée à offrir une excellence perpétuelle dans les expériences logicielles sur le matériel de pointe de NVIDIA avec PyTorch.

En tant que membre essentiel de notre équipe, votre rôle dépassera les frontières traditionnelles, en pilotant l'évolution des charges de travail de formation en apprentissage profond distribué via des technologies de compilation avancées. Vos responsabilités principales incluront :

• Effectuer une analyse approfondie des charges de travail d'apprentissage profond pour identifier et résoudre les goulets d'étranglement de performance, les anomalies fonctionnelles et les inefficacités du système.
• Lier les problèmes de performance directement aux défauts du compilateur ou aux potentiels d'optimisation inexploités et formuler des résolutions robustes.
• Travailler en collaboration pour affiner les transformations de programmes ou en innover de nouvelles à la suite d'examens de performance complets.
• Rester à jour avec les derniers développements dans les compilateurs d'apprentissage profond pour proposer des solutions de pointe qui améliorent l'efficacité du cadre d'apprentissage profond.
• Mener des tests et validations approfondis pour assurer une performance optimale et une qualité dans la formation des modèles.

Venez équipé de :

• Un Bachelor, un Master ou un Ph.D. en informatique ou dans un domaine technique connexe, ou une expérience comparable.
• Une compétence en Python et une familiarité avec les cadres d'apprentissage automatique comme PyTorch ou JAX.
• Une solide compréhension des concepts de compilateur incluant l'interprétation abstraite, les représentations de code telles que la forme SSA et AST, la génération de code, et les transformations de programme.
• Un historique prouvé dans le développement de projets logiciels de taille importante.
• De fortes capacités de communication, tant verbale qu'écrite.

Vous distinguerez avec :

• Des contributions antérieures à des projets de compilateurs d'apprentissage profond open source comme TVM, ou une participation dans des cadres d'apprentissage profond.
• Une connaissance approfondie des internes de PyTorch et/ou JAX.
• Une expertise dans les systèmes distribués, le calcul parallèle et la programmation CUDA.
• Une participation active dans la communauté open source.
• Une expérience de collaboration avec des équipes diversifiées et multidisciplinaires.

Prêt à porter votre carrière à l'avant-garde de l'innovation technologique ? Postulez maintenant pour faire partie de la mission de NVIDIA de pionnier un avenir où l'IA et les compilateurs d'apprentissage profond se réunissent pour créer des applications et des expériences sans précédent.