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

Meta builds technologies that help people connect, find communities, and grow businesses through social technology and immersive experiences like AR and VR.
$117,000 - $173,000
Machine Learning
Staff Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI

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

Meta is seeking an experienced Research Scientist to join their Research & Development teams, specifically focusing on AI Infrastructure. The role is part of two crucial teams: the Kernel team and the E2E Performance team. The Kernel team specializes in maximizing inference performance for Generative AI and Recommendation models through high-performance kernel development, having achieved significant milestones like implementing the first FP8 kernel in Meta's production. The E2E Performance team focuses on optimizing end-to-end performance of AI models using various parallelism strategies and distributed inference techniques.

The ideal candidate will work on cutting-edge problems in AI infrastructure, developing specialized kernels and optimization techniques for large-scale AI models. This position offers the opportunity to work on significant challenges in AI system design and infrastructure efficiency, directly impacting Meta's products and user experiences. The role requires deep expertise in machine learning systems, hardware acceleration, and performance optimization.

The position combines research with practical implementation, requiring both theoretical knowledge and hands-on experience with AI infrastructure and hardware acceleration techniques. You'll be working with state-of-the-art technology in areas like model compression, hardware accelerators, and machine learning compilers. The role offers competitive compensation including base salary, bonus, equity, and comprehensive benefits.

This is an excellent opportunity for someone with a PhD in Computer Science or related field who is passionate about pushing the boundaries of AI infrastructure and system optimization. You'll be part of a team that has already achieved significant breakthroughs, such as enabling AMD GPUs for GenAI production applications, and will continue to drive innovation in the field of AI systems and infrastructure.

Last updated a month ago

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

  • Apply AI infrastructure and hardware acceleration techniques to build & optimize ML systems
  • Develop high performance kernels and parallelism techniques to improve E2E performance
  • Goal setting related to project impact, AI system design, and infrastructure/developer efficiency
  • Deliver impact through deep, thorough data-driven analysis
  • Drive large efforts across multiple teams
  • Define use cases, and develop methodology & benchmarks
  • Apply in-depth knowledge of ML infrastructure system interactions

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

Python
  • PhD degree in Computer Science, Computer Vision, Generative AI, NLP, or relevant technical field
  • Specialized experience in machine learning/deep learning domains
  • Experience developing AI-System infrastructure or AI algorithms in C/C++ or Python
  • Experience with model compression, hardware accelerators architecture, GPU architecture, or machine learning compilers
  • Must obtain work authorization in country of employment

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

Medical Insurance
Vision Insurance
Dental Insurance
  • Bonus
  • Equity
  • Benefits package

Interested in this job?

Jobs Related To Meta Research Scientist, Systems ML - SW/HW Co-Design - Inference

Algorithm Engineer, Machine Learning for Cameras (PhD)

PhD-level Machine Learning Camera Algorithm Engineer position at Meta, focusing on AR/VR innovation and camera technology development.

Technical Program Manager, AI Positions

Technical Program Manager positions at Meta focusing on AI initiatives across research, infrastructure, and product integration.

Software Engineer - Product (Technical Leadership)

Technical leadership software engineering role at Meta focusing on AI infrastructure and product development across major platforms.

Software Engineer, Machine Learning

Senior Machine Learning Engineering role at Meta focusing on developing scalable ML solutions and leading technical initiatives.

Display Image Quality Engineer (PhD)

Display Image Quality Engineer position at Meta focusing on AR/VR display control systems and visual experience optimization.