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.


Equal Employment Opportunity and Affirmative Action Meta is proud to be an Equal Employment Opportunity and Affirmative Action employer. We do not discriminate based upon race, religion, color, national origin, sex (including pregnancy, childbirth, reproductive health decisions, or related medical conditions), sexual orientation, gender identity, gender expression, age, status as a protected veteran, status as an individual with a disability, genetic information, political views or activity, or other applicable legally protected characteristics. You may view our Equal Employment Opportunity notice here.
Meta is committed to providing reasonable support (called accommodations) in our recruiting processes for candidates with disabilities, long term conditions, mental health conditions or sincerely held religious beliefs, or who are neurodivergent or require pregnancy-related support. If you need support, please reach out to accommodations-ext@fb.com.