Research Scientist Intern, PyTorch Distributed (PhD) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Meta · 5 hours ago

Research Scientist Intern, PyTorch Distributed (PhD)

Meta is seeking a Research Scientist Intern to join our Meta PyTorch Distributed Team. The main purpose of this role is to advance the state-of-the-art in machine learning frameworks and to develop novel AI algorithms for large scale distributed training and inference.

Computer Software
check
Comp. & Benefits
badNo H1Bnote

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

PhD in Computer ScienceMachine LearningDeep LearningPyTorchDistributed SystemsGPU ArchitectureHigh Performance ComputingMachine Learning CompilersTeam CollaborationCommunication Skills

Required

Currently has, or is in the process of obtaining, PhD degree in the field of Computer Science or a related STEM field
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
Intent to return to degree program after the completion of the internship/co-op
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

Company

Meta's mission is to build the future of human connection and the technology that makes it possible.

Funding

Current Stage
Late Stage

Leadership Team

leader-logo
Kathryn Glickman
Director, CEO Communications
linkedin
leader-logo
Christine Lu
CTO Business Engineering NA
linkedin
Company data provided by crunchbase