Research Scientist Intern, Systems ML - SW/HW Co-Design - Inference

Meta - Menlo Park, CA

Full-time - Intern

Menlo Park, CA

Web Search Portals, Libraries, Archives, and Other Information Services

About the position

The Research Scientist Intern position in the AI System SW/HW Co-design team at Meta focuses on developing and optimizing high-performance software and hardware technologies for AI at datacenter scale. The role involves working on next-generation AI Systems Inference and Training architectures, collaborating with various hardware types and workload types to drive innovation in low precision numerics, ML operator optimizations, and performance modeling. Interns will engage in cross-functional collaboration to understand infrastructure requirements and influence industry partners to align with Meta's roadmap.

Responsibilities

• Develop tools and methodologies for large scale workload analysis and extract representative benchmarks in C++/Python/Hack.
• Analyze evolving Meta workload trends and business needs to derive requirements for future offerings.
• Utilize extensive understanding of CPUs, GPUs, and systems to identify bottlenecks and enhance product/service efficiency.
• Collaborate closely with software developers to re-architect services and improve codebase through algorithm redesign.
• Identify industry trends and conduct prototyping exercises to quantify the value proposition for Meta.
• Work with Software Services, Product Engineering, and Infrastructure Engineering teams to deliver the hardware roadmap into production.

Requirements

• Currently has, or is in the process of obtaining, a PhD degree in Computer Science or a related STEM field.
• Experience with hardware architecture, compute technologies, and/or storage systems.

Nice-to-haves

• Intent to return to degree-program after the completion of the internship/co-op.
• Track record of achieving results demonstrated by grants, fellowships, patents, or first-authored publications.
• Architectural understanding of CPU, GPU, Accelerators, Networking, and systems.
• Experience with large-scale infrastructure and distributed systems.
• Experience in developing and debugging in C/C++, Python, and/or PyTorch.
• Experience driving original scholarship in collaboration with a team.
• Experience leading a team in solving analytical problems using quantitative approaches.
• Interpersonal experience in cross-group and cross-culture collaboration.
• Experience in theoretical and empirical research.

Benefits

• $7,500/month to $11,333/month compensation
• Equal Employment Opportunity and Affirmative Action employer
• Reasonable accommodations for candidates with disabilities