Compiler Architect, MTIA Software (Technical Leadership)

About the job

Meta is seeking a principal-level Compiler Architect to drive the technical strategy and execution of compiler infrastructure for MTIA (Meta Training and Inference Accelerator). In this role, you will define the compiler architecture that enables efficient code generation, optimization, and execution on Meta's custom AI accelerators. You will tackle the hardest compiler challenges spanning ML workload analysis, graph-level optimizations, memory hierarchy management, and hardware-software co-design. This is a role for engineers who shape the foundational software stack that unlocks the full performance potential of custom silicon for large-scale AI workloads.

Responsibilities

Define and own the compiler architecture and technical roadmap for MTIA, including graph compilers, code generation, and optimization strategies Solve complex compiler optimization challenges spanning operator fusion, memory planning, scheduling, and efficient mapping of ML workloads to custom accelerator hardware Design extensible compiler frameworks and intermediate representations that enable rapid iteration and support evolving ML model architectures Drive performance improvements by identifying and eliminating bottlenecks across the compiler stack, from high-level graph optimizations to low-level code generation Partner with MTIA hardware teams on hardware-software co-design, influencing accelerator architecture decisions based on compiler capabilities and workload requirements Establish compiler correctness, reliability, and performance validation practices that ensure production-quality code generation at scale Collaborate with ML framework teams to ensure seamless integration of MTIA compiler infrastructure with PyTorch and other ML frameworks Evaluate and integrate state-of-the-art compiler technologies such as MLIR, and drive adoption of best practices across the compiler organization Mentor engineers across the organization, leading compiler architecture reviews and establishing a culture of technical excellence in compiler development Communicate complex compiler architecture and strategy clearly to technical and non-technical audiences, producing reference-quality design documents