Compiler Engineer, MTIA Software (Technical...

The Meta Game, Inc.
New York, NY, United States
2 days ago
Apply on www.juju.com
Prepare application

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Compensation
$219,000.0 - $301,000.0
Working hours
Regular working hours
Job source

Tech stack

Clean Code Principles Artificial Intelligence Code Generation Machine Learning Tensorflow Software Engineering Pytorch

Job description

Meta is seeking a principal-level Compiler Architect to work across the full compiler stack for MTIA (Meta Training and Inference Accelerator). In this role, you will design and implement compiler infrastructure that enables efficient code generation, optimization, and execution on Meta’s custom AI accelerators - from high-level graph transformations through intermediate representations to low-level code generation. 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 build the foundational software stack that unlocks the full performance potential of custom silicon for large-scale AI workloads., 1. Design compiler architecture and contribute across the full MTIA compiler stack, including graph compilers, intermediate representations, optimization passes, and code generation

  1. Build extensible compiler frameworks and intermediate representations that enable rapid iteration and support evolving ML model architectures

  2. Identify and eliminate performance bottlenecks across the compiler stack - from high-level graph optimizations through scheduling to low-level code generation

  3. Collaborate with MTIA hardware teams on hardware-software co-design, translating compiler analysis and workload requirements into accelerator architecture improvements

  4. Develop compiler correctness, reliability, and performance validation practices that ensure production-quality code generation at scale

  5. Work with ML framework teams to integrate MTIA compiler infrastructure with PyTorch and other ML frameworks

  6. Evaluate and apply state-of-the-art compiler technologies such as MLIR, bringing best practices into the compiler organization

  7. Mentor engineers across the organization, leading compiler architecture reviews and raising the bar for technical excellence in compiler development

Requirements

  1. Experience leading multi-year cross-functional technical initiatives, including defining metrics, managing dependencies, and driving execution across organizational boundaries

  2. Experience developing high-performance systems software with strong understanding of low-level optimization and hardware architecture

  3. Experience influencing technical direction and engineering practices across multiple teams through written proposals, design reviews, and stakeholder alignment

Preferred Qualifications:

Preferred Qualifications:

  1. Deep understanding of ML model architectures and their computational patterns

  2. Experience adhering to and implementing responsible, ethical AI practices (e.g., risk assessment, bias mitigation, quality and accuracy reviews)

  3. Experience with hardware-software co-design for custom ML accelerators or AI chips

  4. Track record of applying AI tools and automation to redesign engineering workflows, with demonstrated efficiency or quality improvements

  5. Demonstrated ability to integrate AI tools to optimize/redesign workflows and drive measurable impact (e.g., efficiency gains, quality improvements)

  6. Demonstrated ongoing AI skill development (e.g., prompt/context engineering, agent orchestration) and staying current with emerging AI technologies

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on www.juju.com
Prepare application

Good distractions

Talks and stories from around this role — technically off-topic, practically not.

2:35 min

Preventing remote code execution in PyTorch models

Balázs Kiss · World Congress 2023

1:39 min

Fundamentals of tensors and the TensorFlow library

Håkan Silfvernagel · LIVE

2:22 min

Reviewing and troubleshooting code generated by AI assistants

Cassidy Williams · Coffee With Developers

2:08 min

Essential engineering roles in the generative AI space

Mary Grygleski Mary Grygleski · LIVE

1:06 min

Compiling PyTorch environments for advanced time forecasting

Christoph Lohrmann Christoph Lohrmann +1 · World Congress 2026 Europe

1:20 min

Identifying multi-disciplinary talent for developer experience engineering roles

Hazal Mestci +1 · Coffee With Developers

Videos

See all

Related articles

See all