In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger AI and Compute Foundations team. MTIA Software Team has been developing a comprehensive software platform strategy that delivers a highly flexible platform to train & serve current deep learning model architectures, combined with auto-tuned high performance for production environments across specialized hardware architectures.
You will be working on one of the core areas such as PyTorch & GenAI framework components, AI compiler and runtime, high-performance deep learning 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 optimize them efficiently on MTIA hardware. You will also partner with hardware design teams to develop 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.-------À ce poste, vous ferez partie de l'équipe logicielle MTIA (Meta Training & Inference Accelerator) et intégrerez l'équipe plus large chargée de l'IA et des fondements informatiques. L'équipe logicielle de MTIA élabore actuellement une stratégie globale de plate-forme logicielle visant à offrir une plate-forme extrêmement flexible permettant de former et d'exploiter de nouvelles architectures de modèles d'apprentissage profond, tout en garantissant des performances optimisées automatiquement pour les environnements de production sur diverses architectures matérielles spécialisées.




Vous travaillerez sur l'un des domaines clés, tels que les composants des frameworks PyTorch et GenAI, le compilateur et le moteur d'exécution d'IA, les noyaux d'apprentissage profond haute performance et les outils destinés à accélérer les charges de travail d'apprentissage automatique sur les plates-formes matérielles d'IA MTIA actuelles et de nouvelle génération. Vous travaillerez en étroite collaboration avec des chercheurs en IA pour analyser les modèles d’apprentissage profond et les optimiser efficacement sur le matériel MTIA. Vous travaillerez également en collaboration avec les équipes de conception matérielle afin de mettre au point des optimisations visant à atteindre des performances élevées. Vous mettrez en œuvre les meilleures pratiques en matière de développement logiciel pour concevoir des fonctionnalités, optimiser les applications et mettre en œuvre des techniques d'optimisation des performances. Vous bénéficierez d'une expérience précieuse dans le développement de compilateurs d'apprentissage automatique et contribuerez à la conception de logiciels matériels de nouvelle génération pour les problèmes spécifiques du domaine d'IA. Software Engineer, Systems ML
- Frameworks / Compilers / DL-Kernels | Ingénieur logiciel spécialisé en apprentissage automatique des systèmes – cadres/compilateurs/noyaux d'apprentissage profond

Responsibilities:
AI frameworks, compiler stack,



high performance deep learning kernel development, runtime and tooling for MTIA accelerator | Développement d'une pile logicielle dans l'un des domaines clés suivants : Cadres d'IA, pile de compilateurs, développement de noyaux d'apprentissage profond haute performance, environnement d'exécution et outils pour l'accélérateur MTIA. Contribute to the development of the industry-leading PyTorch AI framework, core compilers, and GenAI OSS frameworks to support new state-of-the-art inference and training AI hardware accelerators and optimize their performance | Contribuer au développement du cadre d'IA PyTorch, leader du secteur, des compilateurs de base et des cadres open source GenAI afin de prendre en charge les nouveaux accélérateurs matériels d'IA de pointe destinés à l'inférence et à l'entraînement, et d'optimiser leurs performances. Analyze deep learning models, develop and implement optimization algorithms towards high performance and efficiency gains | Analyser des modèles d'apprentissage profond, développer et mettre en œuvre des algorithmes d'optimisation visant à améliorer les performances et à réaliser des gains d'efficacité. Collaborate with AI research scientists to accelerate the next generation of deep learning models such as recommendation systems and Generative AI | Collaborer avec des chercheurs en intelligence artificielle afin d'accélérer le développement de la prochaine génération de modèles d'apprentissage profond, tels que les systèmes de recommandation et l'IA générative. Proven C/C++ programming skills | Compétences avérées en programmation C/C++ Experience in AI framework development or accelerating deep learning models on hardware architectures | Expérience dans le développement de cadres d

📌 Software Engineer, Systems ML (Canada)
🏢 Meta
📍 Canada