Software Engineer - AI/HPC Systems Specialist

Support the development of Meta's on-house silicon AI accelerators by developing collective communications software library and optimizing distributed AI/ML workloads' performance.

Software Engineer - AI/HPC Systems Specialist Responsibilities:

• Contribute to our developer infrastructure, including simulation and HW emulation platforms, to enable performance measurement and optimization for Meta's in-house accelerator programs.
• Understand and contribute to the collective communications library, intended to be deployed on Meta's AI/ML superclusters.
• Support networking and compute hardware acceleration techniques to improve ML inference and training model performance.
• Perform architectural analysis to ensure system designs meet performance, scalability, and reliability requirements.
• Implement simulation models for Meta's Accelerator ASICs, develop and analyze various scenarios to evaluate data center performance and identify potential improvements.
• Collaborate with architects and engineers to integrate simulation results into system design processes.
• Use instruction set simulators to define performant firmware for Meta's training/inference accelerators.
• Collaborate with hardware and firmware teams to ensure accurate modeling and simulation of accelerator functionalities.
• Analyze simulation results to guide firmware development and optimization efforts.

Minimum Qualifications:

• Bachelor or higher in Computer Science, Computer Engineering, or any other relevant technical field
• 5+ years experience in developing C++ codebase
• 5+ years experience in developing Python codebase
• Understanding of performance and benchmarking measurement and optimization on collective communications and distributed at-scale model training

Preferred Qualifications:

• Full-stack experience and understanding of AI/HPC systems, from HW/infrastructure through the application layer, performance optimizations, including familiarity with relevant tools, libraries, and frameworks (e.g., NCCL, PyTorch, CUDA)
• Experience in one or more of the following machine learning/deep learning domains: hardware accelerators, AI Infrastructure, and/or high performance computing (HPC), particularly pertaining to interconnect and collective
• Knowledge of AI/HPC hardware requirements and specifications (e.g., configuring hardware components, GPU, memory, network for AI/HPC workloads)
• Understanding of the transport stack (e.g., RoCE), its constraints and performance measures and how transport considerations enable the collective communications stack

About Meta:

Meta builds technologies that help people connect, find communities, and grow businesses. When Facebook launched in 2004, it changed the way people connect. Apps like Messenger, Instagram and WhatsApp further empowered billions around the world. Now, Meta is moving beyond 2D screens toward immersive experiences like augmented and virtual reality to help build the next evolution in social technology. People who choose to build their careers by building with us at Meta help shape a future that will take us beyond what digital connection makes possible today-beyond the constraints of screens, the limits of distance, and even the rules of physics.

Individual compensation is determined by skills, qualifications, experience, and location. Compensation details listed in this posting reflect the base hourly rate, monthly rate, or annual salary only, and do not include bonus, equity or sales incentives, if applicable. In addition to base compensation, Meta offers benefits. Learn more about benefits at Meta.