Software Engineer, Systems ML - PyTorch Compiler / ML Framework / Performance

Rejoignez notre équipe de compilateurs PyTorch chez Meta

Aperçu :

L'équipe de compilateur PyTorch chez Meta a pour mission d'améliorer les performances de PyTorch, en se concentrant sur l'efficacité et la vitesse sans compromettre sa flexibilité renommée. Notre technologie révolutionnaire PT2 a établi un nouveau référentiel dans le paysage des cadres ML depuis son lancement en mars 2023, propulsant PyTorch vers le statut de compilateur ML leader de l'industrie. Nous recherchons un ingénieur logiciel pour contribuer au compilateur PT2 et repousser les limites de la technologie d'apprentissage automatique.

Principales responsabilités :

• Développer et améliorer les composants du compilateur PT2 tels que TorchDynamo, TorchInductor et PyTorch Core.
• Conduire des améliorations des performances de PyTorch, fournissant des solutions systématiques bénéficiant à la communauté globale PyTorch.
• Rechercher et optimiser les modèles d'IA générative, en se concentrant sur les étapes de pré-entraînement, de peaufinage et d'inférence.
• Explorer les synergies entre le compilateur PyTorch et les systèmes PyTorch distribués.
• Collaborer activement avec les utilisateurs de PyTorch pour favoriser l'adoption et des utilisations innovantes des technologies PT2.

Qualifications minimales :

• BS/MS en informatique ou dans un domaine étroitement lié.
• Expérience pratique des compilateurs, des systèmes ML, du calcul haute performance et de l'optimisation GPU.
• Maîtrise de la programmation avec Python ou CUDA.
• Poursuivre ou avoir obtenu un diplôme de Bachelor en informatique ou dans un domaine technique connexe.

Qualifications préférées :

• Ph.D. en informatique, mathématiques appliquées, ou domaines connexes.
• Expérience avec les technologies PT2, Triton, MLIR ou contributions directes à PyTorch.
• Expertise approfondie de la performance GPU et du développement de noyaux CUDA haute performance.
• Preuve d'excellence en recherche et en génie logiciel, attestée par des publications, des brevets, ou la participation à des compétitions pertinentes.

À propos de Meta :

Meta continue de redéfinir le paysage de la connectivité et de l'interaction sociale grâce à des technologies innovantes. Depuis le lancement de Facebook en 2004 et les applications subséquentes telles qu'Instagram et WhatsApp, nous nous aventurons maintenant dans la réalité augmentée et virtuelle pour concevoir la prochaine ère de la technologie sociale. Chez Meta, votre travail transcende les frontières digitales traditionnelles, créant de nouvelles façons pour les gens de se connecter, les communautés de prospérer et les entreprises de croître.

Rémunération et avantages :

Le poste offre un salaire compétitif allant de 117 000 à 173 000 dollars par an, en plus des bonus, des actions et des avantages complets. Nous sommes engagés à soutenir et accompagner de manière inclusive les candidats divers dans notre processus de recrutement.

Si vous êtes prêt à façonner l'avenir de la technologie avec nous chez Meta, en faisant avancer l'un des domaines les plus passionnants du monde, postulez dès aujourd'hui pour faire partie de notre équipe de compilateur PyTorch !