Meta is seeking an AI Research Scientist/Software Engineer to join our team. The ideal candidate will have industry experience working on AI infrastructure related topics. The position will involve taking these skills and applying them to solve some of the most crucial and exciting problems that exist on the web.The Collective Compute Enablement team is dedicated to maximizing training performance of Generative AI and Recommendation models on Meta's Training and Inference Accelerator (MTIA). We model and project the performance of current and future training workloads on MTIA while it is being designed to provide early, crucial feedback to the architecture, compiler, and kernels teams. We employ cutting-edge optimization and data parallelization strategies to maximize training throughput for the next generations of LLMs and DLRMs, and we work cross-functionally with many partner teams to assure the end-to-end performance of large-scale training in order to more quickly deliver the next generation of AI experiences to our users.Software Engineer, Systems ML - Collective Compute Enablement ResponsibilitiesApply state-of-the-art AI infrastructure and software/hardware acceleration techniques to build and optimize our large-scale AI workloadsAnalyze, benchmark, and optimize large-scale workloads on next-generation training superclustersDefine use cases and develop methodology and benchmarks to evaluate different approachesSet direction and goals for the team related to project impact, AI system design, infrastructure, and developer efficiencyLead large and complex technical efforts across many engineers and teamsInfluence and impact next-generation of model and hardware architecture choices based on thorough data-driven analysesHelps onboard new team members, provides mentorship, and enables successful ramp up on the team's code baseMentor other engineers, research scientists and improve the quality of engineering work in the broader teamMinimum QualificationsBachelor’s degree in computer science or a related STEM fieldSpecialized experience in one or more of the following machine learning/AI domains: hardware accelerator architectures, machine learning compilers or ML systems, AI infrastructure, high-performance computing, performance optimizations or ML frameworks such as PyTorchProven C/C++ and Python programming skills in developing AI Systems infrastructure or AI algorithmsExperience with debugging in C++, Python and/or PyTorchTrack record of defining and leading long-term plans for the teamTrack record of mentoring and growing other engineersMust obtain work authorization in the country of employment at the time of hire, and maintain ongoing work authorization during employmentTechnical leadership experiencePreferred QualificationsMaster's degree/PhD in computer science or related STEM field and experience in AI framework development or accelerating AI models on hardware architecturesExperience with training of large-scale AI modelsExperience with distributed AI systems and communication protocols such as MPI or collective libraries such as NCCLExperience or knowledge in one or more of LLM/LDM, ranking and recommender models or collective communication librariesAbout 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 Meta is proud to be an Equal Employment Opportunity 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 accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures. If you need assistance or an accommodation due to a disability, fill out the Accommodations request form.