Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Meta is seeking a Software Engineer for their MTIA (Meta Training & Inference Accelerator) Software team within the PyTorch AI framework organization. This role focuses on developing comprehensive AI Compiler strategies and high-performance solutions for ML workloads. The position involves working with cutting-edge AI technologies, including PyTorch framework components, AI compiler and runtime systems, and high-performance kernels.

The role offers an opportunity to work directly with AI researchers and hardware design teams, implementing compiler optimizations and developing solutions for next-generation AI hardware platforms. You'll be part of a team that directly impacts the performance and deployment of AI training and inference platforms at Meta.

Key areas of focus include developing compiler stack solutions, optimizing deep learning frameworks, and creating high-performance kernels for specialized hardware architectures. The position requires strong programming skills in C/C++, experience with AI frameworks, and understanding of hardware architectures.

This is an excellent opportunity for someone passionate about AI infrastructure, compiler optimization, and high-performance computing. You'll be working at the intersection of hardware and software, helping to shape the future of AI acceleration at Meta. The role offers competitive compensation, including base salary, bonus, equity, and comprehensive benefits.

The ideal candidate will have experience in AI framework development, compiler optimizations, or high-performance kernel development, with a strong academic background in Computer Science or related fields. You'll be contributing to Meta's mission of building technologies that help people connect and pushing the boundaries of AI infrastructure development.