Meta · 3 hours ago
Research Scientist Intern, PyTorch Distributed (PhD)
Menlo Park, CA
Internship
Onsite
Intern
$7,650/mo - $12,134/moMeta is seeking a Research Scientist Intern to join their Meta PyTorch Distributed Team. The role involves applying AI and machine learning techniques to enhance the PyTorch framework and collaborating with users to enable new use cases for the framework.
Computer Software
Responsibilities
Apply relevant AI and machine learning techniques to advance the state-of-the-art in machine learning frameworks
Collaborate with users of PyTorch to enable new use cases for the framework both inside and outside Meta
Develop novel, accurate AI algorithms and advanced systems for large scale distributed training and inference
Leverage graph-based and compiler-based technologies to optimize distributed training and distributed inference use-cases
Qualification
Required
Experience in one or more of the following machine learning/deep learning domains: Large scale training and inference ML Systems Research, ML theory: Basic knowledge about ML models in different modalities like LLM (Large Language Models), Vision (VITS, MVITS) and Multimodal and how scale impacts performance, ML systems: AI infrastructure, machine learning accelerators, high performance computing, machine learning compilers, GPU architecture, machine learning frameworks, distributed systems, on-device optimization
Must obtain work authorization in the country of employment at the time of hire, and maintain ongoing work authorization during employment
Preferred
Experience or knowledge on training models at scale using PyTorch/TensorFlow/JAX
Experience or knowledge on working with a distributed GPU cluster
Proven track record of achieving significant results as demonstrated by grants, fellowships, patents, as well as first-authored publications at leading workshops or conferences such as NeurIPS, MLSys, ASPLOS, PLDI, CGO, PACT, ICML, or similar
Experience working and communicating cross functionally in a team environment
Benefits
Benefits
Company
Meta
Meta's mission is to build the future of human connection and the technology that makes it possible.
Founded in 2004
10001+ employees
Funding
Current Stage
Late StageLeadership Team
Kathryn Glickman
Director, CEO Communications
Christine Lu
CTO Business Engineering NA
Recent News
Crunchbase News
2025-11-17
2025-11-16
Company data provided by crunchbase