Manager, Software Engineering, AI Hardware Acceleration

Posted 19 days ago
Main Location
Menlo Park, CA, United States
Open jobs

Facebook builds systems that allow more than a billion people all over the world to connect and communicate using whatever devices they have available. Our systems need to be scalable, fast, reliable, and efficient in order to provide a great user experience. An end-to-end approach is required which includes integration between frameworks, systems, compilers, and HW architecture in order to execute state-of-the-art machine learning using hardware acceleration. The E2E stack is crucial that can affect performance, efficiency and deployment velocity in both training and inference. In order to further the state-of-the-art in AI/ML, the AI HW Accelerator team, has been developing a comprehensive PyTorch AI Compiler backend, Runtime, Programming Model that delivers a highly flexible platform to explore new model architectures, combined with auto-tuned high performance for production environments across a wide range of hardware architectures. We actively engage with the community through publications, open source software, participation in technical conferences and workshops, and collaborations with academia. Facebook AI researchers and engineers work from our offices around the globe.

  • Manage a team of domain experts that builds the PyTorch AI Compiler, Runtime.
  • Create and optimize compiler backend to leverage the full hardware potentials. Take a principle approach for targeting new architectures and efficiently optimizing them such as Tensor Expression, Polyhedral approaches.
  • Integrate with PyTorch and optimize for AI chips/Accelerators.
  • Develop tests, test plans, and testing infrastructure to verify new/existing algorithms, workloads and functionality of the accelerator and their performance.
  • Develop customized high-performance computation library for AI chip.
  • Drive and measure KPI’s as part of Silicon enablement: Perf/W
  • Perf/$, Perf etc.
  • Manage engineers and help them develop their careers, assigning them to projects tailored to their skill levels, long-term skill development, personalities, and work styles.
  • Communicate and collaborate effectively with cross-functional hardware and software engineering teams.
  • Champion engineering and operational excellence, establishing metrics and process for regular assessment and improvement.
  • Set clear expectations and create a positive work environment based on accountability, in collaboration with the engineering and management teams.
  • Operate strategically and tactically. Develop the vision and strategy to help set direction for the team, while staying on top of the day-to-day software development.
  • Work closely with dedicated recruiting staff to expand the team, including sourcing candidates, interviewing candidates, participating in conferences/events, and on-boarding new employees.

MINIMUM QUALIFICATIONS

  • Experience operating in a fast-moving environment where the workloads evolve at a rapid pace.
  • Experience managing a team of compiler engineers, ASIC Software and Firmware bringup teams.
  • Demonstrated experience recruiting, building, structuring, leading technical organizations, including performance management.
  • Understand and analyze the interplay of hardware and software architectures on future algorithms, programming models and applications.
  • Develop innovative architectures to extend the state-of-the-art in DL performance and efficiency.
  • Experience with CPUs, GPUs and accelerators.
  • Experience with Hardware and Software Co-design.
  • Basic understanding of Compilers, LLVM or machine learning compilers.
  • Expertise with AI compiler backend optimizations (TVM, XLA etc.), backend code generators.
  • Prototype key deep learning, data analytics algorithms and applications.
  • Experience driving power-performance trades.
  • Familiarity with Video, DSP, PCIe, SoC Interconnect and NoC.

PREFERRED QUALIFICATIONS

  • Experience operating in a fast-moving environment where the workloads evolve at a rapid pace.
  • Experience managing a team of compiler engineers, ASIC Software and Firmware bringup teams.
  • Demonstrated experience recruiting, building, structuring, leading technical organizations, including performance management.
  • Understand and analyze the interplay of hardware and software architectures on future algorithms, programming models and applications.
  • Develop innovative architectures to extend the state-of-the-art in DL performance and efficiency.
  • Experience with CPUs, GPUs and accelerators.
  • Experience with Hardware and Software Co-design.
  • Basic understanding of Compilers, LLVM or machine learning compilers.
  • Expertise with AI compiler backend optimizations (TVM, XLA etc.), backend code generators.
  • Prototype key deep learning, data analytics algorithms and applications.
  • Experience driving power-performance trades.
  • Familiarity with Video, DSP, PCIe, SoC Interconnect and NoC.

Facebook's mission is to give people the power to build community and bring the world closer together. Through our family of apps and services, we're building a different kind of company that connects billions of people around the world, gives them ways to share what matters most to them, and helps bring people closer together. Whether we're creating new products or helping a small business expand its reach, people at Facebook are builders at heart. Our global teams are constantly iterating, solving problems, and working together to empower people around the world to build community and connect in meaningful ways. Together, we can help people build stronger communities - we're just getting started.

Mission
We're a community of women leveraging our connections into top companies to help underrepresented women get the roles they've always deserved. Simultaneously, we work to build truly inclusive hiring processes and environments where women can thrive and not just survive.
Are you hiring? Join our platform for diversifiying your team
Manager, Software Engineering, AI Hardware Acceleration
Facebook, Inc.