Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Meta is seeking a Senior Software Engineer to join their MTIA (Meta Training & Inference Accelerator) Software team within the PyTorch AI framework organization. This role focuses on developing cutting-edge AI compiler technologies and optimizations for machine learning workloads.

The position involves working on core areas including PyTorch framework components, AI compiler and runtime systems, and high-performance kernels for current and next-generation AI hardware platforms. You'll collaborate closely with AI researchers to analyze and optimize deep learning models for MTIA hardware, while partnering with hardware design teams to develop compiler optimizations for maximum performance.

As a member of this team, you'll be at the forefront of AI infrastructure development, applying software best practices to design features and optimization techniques. The role offers valuable experience in developing machine learning compiler frameworks and driving next-generation hardware-software co-design for AI-specific challenges.

Key responsibilities include developing the SW stack with focus on AI frameworks and compiler optimization, contributing to PyTorch's core compilers, analyzing deep learning networks, and collaborating with research scientists to accelerate various ML models including recommendation systems, generative AI, computer vision, and NLP.

The ideal candidate should have strong C/C++ programming skills, experience with AI frameworks or hardware acceleration, and a relevant technical degree. Knowledge of GPU/CPU architectures, experience with frameworks like PyTorch/TensorFlow, and expertise in compiler optimizations or high-performance computing are highly valued.

This role offers competitive compensation (CA$133,000-176,000/year) plus bonus and equity, along with comprehensive benefits. Join Meta's team in Toronto to help shape the future of AI infrastructure and machine learning optimization.