Software Engineering Manager, AI Compiler

$177,000 - $251,000/Yr

Meta - Sunnyvale, CA

Full-time - Mid Level

Sunnyvale, CA

Web Search Portals, Libraries, Archives, and Other Information Services

About the position

The Software Engineering Manager for AI Compiler at Meta will lead a team of engineers focused on developing a compiler and runtime toolchain for advanced deep learning hardware components tailored for augmented and virtual reality systems. This role involves strategic leadership, fostering collaboration across teams, and driving the development of a robust software development kit (SDK) for custom machine learning accelerators.

Responsibilities

• Grow a team of domain experts in ML compilation for deep learning hardware acceleration targets.
• Communicate, collaborate, and build relationships with clients and peer teams to facilitate cross-functional projects.
• Operate strategically and tactically.
• Develop vision, strategy, and help set direction for the team.
• Remain up to date with ongoing software development activities in the team, help work through technical challenges, and be involved in design decisions.
• Build a collaborative, high-performing team culture focused on delivering outcomes against aggressive goals.

Requirements

• Bachelor's or above in Computer Science or Electronics Engineering or an equivalent field.
• Experience with software toolchain development for custom hardware, such as GPUs, AI accelerators, or other hardware accelerators.
• 2+ years of experience in leading a team of software engineers of varied skill levels.
• Experience with cross-functional collaboration with hardware or AI framework teams.
• Knowledge of computer algorithms and experience crossing multi-disciplinary boundaries to drive optimal system solutions.
• Experience with machine-code generation / compiler back-ends.

Nice-to-haves

• Masters or above in Computer Science or Electronics Engineering or an equivalent field.
• Experience in accelerating libraries on AI hardware, similar to cuBLAS, cuDNN, CUTLASS, HIP, ROCm, etc.
• Experience working on and contributing to an active compiler toolchain codebase, such as LLVM, MLIR, GCC, MSVC, Glow.
• Experience in deep learning algorithms and techniques, e.g., convolutional neural networks, recurrent networks, LLMs, etc.
• Experience with a mainstream machine-learning framework, e.g. PyTorch, Tensorflow or Caffe.

Benefits

• Competitive salary ranging from $177,000 to $251,000 per year.
• Bonus opportunities.
• Equity options.
• Comprehensive benefits package.