Join Our Team: Software Engineering Manager for ML/AI Frameworks at Meta

At Meta, we're pioneering the path to the next technological revolution. The Meta Training & Inference Accelerator (MTIA) Software team is at the forefront, enhancing AI frameworks to boost the efficiency of Meta’s deep learning and machine learning operations. Utilizing our specialized MTIA AI accelerator hardware, we ensure high performance and adaptability.

We are currently seeking a Software Engineering Manager to lead our team of kernel engineers. This role focuses on driving progress and innovation in our deep learning kernel libraries, tailored specifically for the MTIA AI accelerator hardware. If you're ready to lead a cutting-edge team in a dynamic and transformative industry, we want to hear from you!

Key Responsibilities:

• Expand and guide a team of experts specialized in high-performance deep learning kernel development.
• Enhance collaboration and relationships with clients and peer teams, ensuring the success of cross-functional projects.
• Strategically develop the vision and direction for the team while navigating day-to-day tactical challenges.
• Maintain awareness of software development activities within your team, assist in addressing technical hurdles, and participate in design decisions.

Minimum Qualifications:

• Proven experience in deep learning kernel development using CPUs, GPUs, or AI accelerators.
• Over two years of managerial experience leading diverse teams of kernel engineers.
• Robust experience in cross-functional collaboration with hardware or AI framework/compiler teams.
• A track record of effective recruitment, team building, organizational structuring, and performance management.

Preferred Qualifications:

• Expertise in accelerating libraries on AI hardware such as cuBLAS, cuDNN, CUTLASS, HIP, ROCm, etc.
• Experience with various high-performance computing programming models like GPU CUDA programming, OpenCL, or OpenMP.
• Familiarity with hardware architectures including Intel SIMD, GPUs, RISC-V, and ML Accelerators.
• Knowledge in a hardware-software integrated development environment, including simulators and FPGA emulators.
• Proficiency in major ML frameworks like PyTorch, TensorFlow, ONNX, MXNet, etc.

Meta is dedicated to finding innovative ways to bridge distances and bring people together—from enhancing everyday interactions with augmented and virtual reality to providing robust support in our hiring processes for diverse applicants.

Compensation and Benefits:

The role offers a competitive salary range from $177,000 to $251,000/year, plus bonuses, equity, and comprehensive benefits. Your individual pay will be based on your skills, qualifications, and experience.

If you are eager to contribute to shaping the future of technology with Meta, apply today to become a pivotal part of our journey as a Software Engineering Manager for ML/AI Frameworks.

Meta is committed to providing an inclusive and welcoming environment for all members of our community. Applicants in need of special assistance or accommodation during the interview process or in accessing our website may contact [email protected] for assistance.