Back to jobsJob overview

About the role

Software Engineering Manager - Meta Superintelligence Labs - Infra: Optimizations Team at Meta

Required Skills

cudapytorchai/mldistributed systemsgpu optimizationkernel developmentgenaisystems architecture

About the Role

Meta seeks a hands-on Software Engineering Manager for the Superintelligence Labs Infra Optimizations team. The role involves leading a team to develop high-performance kernels and optimizations for next-generation GenAI models like Llama. Responsibilities include technical management, enabling team growth, and working cross-functionally across hardware and software teams.

Key Responsibilities

  • Lead and support the team developing kernels (e.g., GEMMs, Attention) for GenAI models
  • Enable growth of individual contributors and drive the technical roadmap
  • Lead a high-performance team to deliver efficient compute systems for the fleet
  • Technical management and cross-functional collaboration across hardware and software teams

Required Skills & Qualifications

Must Have:

  • MS or BS in Computer Science, Electrical/Electronics Engineering, or equivalent
  • 3+ years of experience directly managing or leading a team of engineers
  • Experience in leading teams on high performance computing (HPC) and AI/ML systems
  • Experience with GPU/ASIC-based kernel development and optimization (e.g., CUDA), distributed systems, systems architecture, and large-scale programs

Nice to Have:

  • Familiarity with optimizing GenAI workloads and frameworks like PyTorch, TorchTriton
  • Understanding of kernel enablement, optimizations, and attention kernels
  • Knowledge of GPU memory hierarchy, low-level CUDA optimizations, quantization, and sparsity
  • Experience with optimizers such as Adam, Shampoo, Muon

Benefits & Perks

  • Bonus
  • Equity
  • Benefits package