Job Description

Toronto, ON o Full Time

Meta

AI Infrastructure

Artificial Intelligence

In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new DL/ML model architectures, combined with auto-tuned high performance for production environments across specialized hardware architectures. The compiler stack, DL graph optimizations, and kernel authoring for specific hardware, directly impacts performance and deployment velocity of both AI training and inference platforms at Meta.

You will be working on one of the core areas such as PyTorch framework components, AI compiler and runtime, high-performance kernels and tooling to accelerate machine learning workloads on the current & next generation of MTIA AI hardware platforms. You will work closely with AI researchers to analyze deep learning models and lower them efficiently on MTIA hardware. You will also partner with hardware design teams to develop compiler optimizations for high performance. You will apply software development best practices to design features, optimization, and performance tuning techniques. You will gain valuable experience in developing machine learning compiler frameworks and will help in driving next generation hardware software codesign for AI domain specific problems.
-
-
Dans ce role, vous serez membre de l'equipe logicielle MTIA et ferez partie de la plus grande organisation du cadre d'IA PyTorch, a la pointe de l'industrie. L'equipe logicielle de MTIA a developpe une strategie complete de compilateur d'IA qui offre une plateforme tres flexible pour former et fournir de nouvelles architectures de modeles DL/ML, combinee a des performances elevees autoregulees pour les environnements de production sur des architectures materielles specialisees. La pile de compilateurs, les optimisations de graphes DL et la creation de noyaux pour un materiel specifique ont un impact direct sur les performances et la vitesse de developpement des plates-formes d'apprentissage et d'inference d'IA chez Meta.

Vous travaillerez sur l'un des domaines cles tels que les composants du cadre PyTorch, le compilateur et le moteur d'execution de l'IA, les noyaux de haute performance et l'outillage pour ameliorer les charges de travail d'apprentissage automatique sur les plates-formes materielles d'IA actuelles et de la prochaine generation de MTIA. Vous travaillerez en etroite collaboration avec les chercheurs en IA pour analyser les modeles d'apprentissage profond et les reduire efficacement sur le materiel MTIA. Vous vous associerez egalement aux equipes de conception de materiel pour developper des optimisations de compilateurs pour obtenir des performances elevees. Vous appliquerez les meilleures pratiques de developpement logiciel pour concevoir des fonctionnalites, des techniques d'optimisation et de reglage des performances. Vous beneficierez d'une experience precieuse dans le developpement de compilateurs d'apprentissage automatique et contribuerez a la conception de logiciels materiels de nouvelle generation pour les problemes specifiques du domaine d'IA.
-

Software Engineer, Systems ML - Frameworks / Compilers / Kernels | Ingenieur logiciel, Systemes ML, cadres/Compilateurs/Noyaux Responsibilities

+ Development of SW stack with one of the following core focus areas: AI frameworks, compiler stack, high performance kernel development and acceleration onto next generation of hardware architectures. | Developpement d'une pile de logiciels dans l'un des domaines fondamentaux suivants : Cadre d'IA, pile de compilateurs, developpement de noyaux a haute performance et integration dans les architectures materielles de la prochaine generation.
+ Contribute to the development of the industry-leading PyTorch AI framework core compilers to support new state of the art inference and training AI hardware accelerators and optimize their performance. | Contribuer au developpement des compilateurs de base du cadre d'IA PyTorch, leader sur le marche, afin de prendre en charge les nouveaux integrateurs materiels d'IA de pointe en matiere d'inference, d'entrainement et d'optimiser leurs performances.
+ Analyze deep learning networks, develop & implement compiler optimization algorithms. | Analyse des reseaux d'apprentissage profond, developpement et mise en oeuvre d'algorithmes d'optimisation des compilateurs.
+ Collaborating with AI research scientists to accelerate the next generation of deep learning models such as Recommendation systems, Generative AI, Computer vision, NLP etc. | Collaborer avec des chercheurs en IA pour ameliorer la prochaine generation de modeles d'apprentissage profond tels que les systemes de recommandation, l'IA generative, la vision par ordinateur, le traitement automatique des langues, etc.
+ Performance tuning and optimizations of deep learning framework & software components. | Optimisation des performances du cadre d'apprentissage profond et des composants logiciels.

Minimum Qualifications

+ Proven C/C++ programming skills | Competences averees en programmation C/C++
+ Currently has, or is in the process of obtaining a Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience. Degree must be completed prior to joining Meta. | Possede, ou est en train d'obtenir une licence en informatique, en genie informatique, dans un domaine technique pertinent, ou une experience pratique equivalente. Le diplome doit etre obtenu avant de rejoindre Meta.
+ Experience in AI framework development or accelerating deep learning models on hardware architectures. | Experience dans le developpement des cadres d'IA ou dans l'amelioration des modeles d'apprentissage profond sur des architectures materielles.

Preferred Qualifications

+ A Bachelor's degree in Computer Science, Computer Engineering, relevant technical field and 4+ years of experience in AI framework development or accelerating deep learning models on hardware architectures OR a Master's degree in Computer Science, Computer Engineering, relevant technical field and 2+ years of experience in AI framework development or accelerating deep learning models on hardware architectures OR a PhD in Computer Science Computer Engineering, or relevant technical field. | Un baccalaureat en informatique, en ingenierie informatique ou dans un domaine technique pertinent et plus de quatre ans d'experience dans le developpement de cadres d'IA ou l'amelioration des modeles d'apprentissage profond sur des architectures materielles OU une maitrise en informatique, en ingenierie informatique ou dans un domaine technique pertinent et plus de deux ans d'experience dans le developpement de cadres d'IA ou l'amelioration des modeles d'apprentissage profond sur des architectures materielles OU un doctorat en informatique, en ingenierie informatique ou dans un domaine technique pertinent.
+ Knowledge of GPU, CPU, or AI hardware accelerator architectures. | Connaissance des architectures d'accelerateurs materiels de type processeur graphique, unite centrale ou IA.
+ Experience working with frameworks like PyTorch, Caffe2, TensorFlow, ONNX, TensorRT | Experience de travail avec des cadres comme PyTorch, Caffe2, TensorFlow, ONNX, TensorRT
+ OR AI high performance kernels: Experience with CUDA programming, OpenMP / OpenCL programming or AI hardware accelerator kernel programming. Experience in accelerating libraries on AI hardware, similar to cuBLAS, cuDNN, CUTLASS, HIP, ROCm etc. | Noyaux a haute performance pour l'IA : Experience dans la programmation CUDA, OpenMP/OpenCL ou dans la programmation du noyau d'un accelerateur de materiel d'IA. Experience dans l'acceleration de bibliotheques sur du materiel d'IA, comme cuBLAS, cuDNN, CUTLASS, HIP, ROCm, etc.
+ OR AI Compiler: Experience with compiler optimizations such as loop optimizations, vectorization, parallelization, hardware specific optimizations such as SIMD. Experience with MLIR, LLVM, IREE, XLA, TVM, Halide is a plus. | OU compilateur d'IA : Experience dans l`optimisation des compilateurs tels que l`optimisation des boucles, la vectorisation, la parallelisation, l`optimisation specifique du materiel tel que SIMD. L'experience avec MLIR, LLVM, IREE, XLA, TVM, Halide est un plus.
+ OR AI frameworks: Experience in developing training and inference framework components. Experience in system performance optimizations such as runtime analysis of latency, memory bandwidth, I/O access, compute utilization analysis and associated tooling development. | OU cadres d'IA : Experience dans le developpement de composants de formation et du cadre d'inference. Experience dans l'optimisation des performances systemes, telle que l'analyse de la latence, de la largeur de bande de la memoire, de l'acces aux E/S, de l'analyse de l'utilisation du calcul et du developpement d'outils associes.

For those who live in or expect to work from California if hired for this position.

About Meta

Meta builds technologies that help people connect, find communities, and grow businesses. When Facebook launched in 2004, it changed the way people connect. Apps like Messenger, Instagram and WhatsApp further empowered billions around the world. Now, Meta is moving beyond 2D screens toward immersive experiences like augmented and virtual reality to help build the next evolution in social technology. People who choose to build their careers by building with us at Meta help shape a future that will take us beyond what digital connection makes possible today--beyond the constraints of screens, the limits of distance, and even the rules of physics.




Meta developpe des technologies qui aident les gens a se connecter, a trouver des communautes et a developper des entreprises. Lorsque Facebook a ete lance en 2004, il a change la facon dont les gens se connectent. Des applications comme Messenger, Instagram et WhatsApp ont permis a des milliards de personnes dans le monde de s'emanciper. Aujourd'hui, Meta depasse les ecrans 2D pour se tourner vers des experiences immersives comme la realite augmentee et virtuelle, afin de contribuer a la prochaine evolution de la technologie sociale. Les personnes qui choisissent de batir leur carriere en oeuvrant avec nous chez Meta contribuent a faconner un avenir qui nous menera au-dela de ce que la connexion numerique rend possible aujourd'hui -- au-dela des contraintes des ecrans, des limites de la distance, et meme des regles de la physique.




CA$104,000/year to CA$148,000/year + bonus + equity + benefits



Individual compensation is determined by skills, qualifications, experience, and location. Compensation details listed in this posting reflect the base hourly rate, monthly rate, or annual salary only, and do not include bonus, equity or sales incentives, if applicable. In addition to base compensation, Meta offers benefits. Learn more about benefits at Meta.

Equal Employment Opportunity

Meta is proud to be an Equal Employment Opportunity employer. We do not discriminate based upon race, religion, color, national origin, sex (including pregnancy, childbirth, reproductive health decisions, or related medical conditions), sexual orientation, gender identity, gender expression, age, status as a protected veteran, status as an individual with a disability, genetic information, political views or activity, or other applicable legally protected characteristics.


Meta is committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures.




Meta est fiere d'etre un employeur offrant l'egalite des chances en matiere d'emploi. Nous ne faisons aucune discrimination fondee sur la race, la religion, la couleur, l'origine nationale, le sexe (y compris la grossesse, l'accouchement, les decisions en matiere de sante reproductive ou les problemes medicaux connexes), l'orientation sexuelle, l'identite de genre, l'expression de genre, l'age, le statut d'ancien combattant protege, le statut de personne handicapee, les informations genetiques, les opinions politiques ou les activites, ou d'autres caracteristiques protegees par la loi applicables. Vous pouvez consulter notre avis sur l'egalite des chances en matiere d'emploi ici.




Meta s'engage a fournir des amenagements raisonnables aux personnes handicapees et aux anciens combattants invalides dans le cadre de ses procedures de candidature. Si vous avez besoin d'aide ou d'une mesure d'adaptation en raison d'un handicap, remplissez le formulaire de demande d'adaptation.