Software Engineer, Systems ML - Frameworks / Compilers / Kernels | Ingénieur logiciel, Systèmes[...]

Meta; Toronto; On-site; CAD 80,000 - 130,000; Full time

Job summary

An innovative industry leader is seeking a Software Engineer to join their cutting-edge team focused on AI compiler development. In this role, you will work with the renowned PyTorch framework, enhancing machine learning workloads and optimizing performance across specialized hardware. This position offers a unique opportunity to collaborate with top AI researchers and contribute to groundbreaking advancements in hardware-software co-design for AI. If you are passionate about machine learning and eager to make a significant impact in the field, this role is perfect for you.

Benefits

Health Insurance

Flexible Working Hours

Professional Development Opportunities

Remote Work Options

Wellness Programs

Qualifications

• Experience in developing machine learning compiler frameworks.
• Strong understanding of AI and deep learning model optimization.

Responsibilities

• Develop and optimize components of the PyTorch framework.
• Collaborate with hardware design teams for AI solutions.

Skills

PyTorch

Machine Learning

AI Compiler Development

High-Performance Computing

Collaboration with AI Researchers

Education

Bachelor's Degree in Computer Science or related field

Master's Degree in a relevant area

Tools

Machine Learning Frameworks

Compiler Tools

Performance Optimization Tools

Job description

Software Engineer, Systems ML - Frameworks / Compilers / Kernels | Ingénieur logiciel, Systèmes ML, cadres/Compilateurs/Noyaux

In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the industry-leading PyTorch AI framework organization. The team develops a comprehensive AI Compiler strategy that provides a flexible platform for training and serving new DL/ML model architectures, with auto-tuned high performance for production across specialized hardware architectures. Your work will focus on core areas such as PyTorch framework components, AI compiler and runtime, high-performance kernels, and tooling to accelerate machine learning workloads on current and next-generation MTIA hardware platforms. Collaboration with AI researchers and hardware design teams will be essential to analyze deep learning models, optimize compilers, and develop high-performance solutions. This role offers valuable experience in developing machine learning compiler frameworks and contributes to the next generation of hardware-software co-design for AI.

Dans ce rôle, vous serez membre de l'équipe logicielle MTIA et ferez partie de la plus grande organisation du cadre d'IA PyTorch, à la pointe de l'industrie. L'équipe logicielle de MTIA a développé une stratégie complète de compilateur d'IA qui offre une plateforme très flexible pour former et fournir de nouvelles architectures de modèles DL/ML, combinée à des performances élevées autorégulées pour les environnements de production sur des architectures matérielles spécialisées. La pile de compilateurs, les optimisations de graphes DL et la création de noyaux pour un matériel spécifique ont un impact direct sur les performances et la vitesse de développement des plates-formes d'apprentissage et d'inférence d'IA chez Meta. Vous travaillerez sur l'un des domaines clés tels que les composants du cadre PyTorch, le compilateur et le moteur d'exécution de l'IA, les noyaux de haute performance et l'outillage pour améliorer les charges de travail d'apprentissage automatique sur les plates-formes matérielles d'IA actuelles et de la prochaine génération de MTIA. Vous collaborerez étroitement avec des chercheurs en IA et des équipes de conception matérielle pour analyser, optimiser et développer des solutions performantes. Cette position offre une expérience précieuse dans le développement de cadres de compilation pour l'apprentissage automatique et contribue à la conception logicielle matérielle de nouvelle génération pour les défis spécifiques du domaine de l'IA.