Software Engineering Manager, AI Compiler

$177,000 - $251,000/Yr

Meta - Austin, TX

Full-time - Mid Level

Austin, TX

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 compiler and machine learning framework engineers to develop a compiler and runtime toolchain for advanced deep learning hardware components tailored for augmented and virtual reality systems. This role involves strategic leadership, fostering a collaborative team culture, and driving the development of a robust software development kit (SDK) for custom ML 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.