Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Meta's MTIA (Meta Training & Inference Accelerator) Software team is seeking a skilled Software Engineer to join their PyTorch AI framework organization. This role focuses on developing comprehensive AI Compiler strategies and high-performance solutions for ML workloads. You'll be working on core areas including PyTorch framework components, AI compiler and runtime, and high-performance kernels for MTIA AI hardware platforms.

The position offers an opportunity to work directly with AI researchers in analyzing and optimizing deep learning models, while also collaborating with hardware design teams on compiler optimizations. You'll be instrumental in developing machine learning compiler frameworks and driving next-generation hardware-software codesign for AI applications.

The ideal candidate should have strong expertise in C/C++ programming and significant experience in AI framework development. Knowledge of GPU architectures and experience with major ML frameworks like PyTorch and TensorFlow is highly valued. The role requires either 8+ years of experience with a Master's degree or 7+ years with a PhD in Computer Science or related fields.

Working at Meta, you'll be part of a company that's shaping the future of social technology, moving beyond traditional 2D screens toward immersive AR/VR experiences. The position offers competitive compensation, including base salary, bonuses, equity, and comprehensive benefits. You'll be contributing to Meta's mission of connecting people and building communities while working on cutting-edge AI infrastructure that powers next-generation technologies.