Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Meta is seeking a Software Engineer for their MTIA (Meta Training & Inference Accelerator) Software team, which is part of the industry-leading PyTorch AI framework organization. This role focuses on developing cutting-edge AI compiler strategies and platforms for training and serving deep learning/machine learning model architectures. The position involves working with core PyTorch framework components, AI compilers and runtime, and high-performance kernels to accelerate machine learning workloads on current and next-generation MTIA AI hardware platforms.

The role requires close collaboration with AI researchers to analyze and optimize deep learning models for MTIA hardware implementation. You'll partner with hardware design teams to develop compiler optimizations for enhanced performance. The position demands expertise in software development best practices, feature design, and performance tuning techniques.

This is an excellent opportunity for experienced engineers passionate about machine learning infrastructure and compiler frameworks. You'll be at the forefront of developing next-generation hardware-software solutions for AI-specific challenges. The role offers competitive compensation (CA$133,000-176,000/year) plus bonus and equity, reflecting Meta's commitment to attracting top talent in the AI/ML space.

The ideal candidate should have strong C/C++ programming skills, experience with AI frameworks or deep learning model acceleration, and a relevant technical degree. Knowledge of GPU architectures and experience with major ML frameworks like PyTorch, TensorFlow, and ONNX are highly valued. This position offers the chance to work on transformative AI technologies while contributing to Meta's industry-leading AI infrastructure.