Research Scientist Intern, PyTorch Compiler (PhD)

Job Description

The Research Scientist Intern position at Meta focuses on advancing PyTorch technologies, particularly in compiler and distributed systems. Interns will develop new techniques, optimize AI models, and collaborate with users to enhance PyTorch's performance and capabilities.

Requirements

Currently has or is in the process of obtaining a PhD degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience
Experience in ML compiler, Distributed Training, ML systems, or similar
Proficient in Python or Cuda programming
Must obtain work authorization in the country of employment at the time of hire, and maintain ongoing work authorization during employment

Responsibilities

Develop new techniques in TorchDynamo, TorchInductor, PyTorch core, PyTorch Distributed
Explore the intersection of PyTorch compiler and PyTorch Distributed
Optimize Generative AI models across the stack (pre-training, fine-tuning, and inference)
Improve general PyTorch performance
Conduct cutting-edge research on ML compiler and ML distributed technologies
Collaborate with users of PyTorch to enable new use cases for the framework both inside and outside Meta

Benefits

Employees at Meta are often offered comprehensive benefits, including medical/dental/vision coverage, mental-health resources, family planning support (fertility, adoption, surrogacy), and caregiving programs. Financial benefits typically include a competitive retirement plan, equity awards, life insurance, and access to financial coaching and legal support. Time-off benefits commonly feature generous PTO and holidays, unlimited sick time, various paid leaves, flexible “Global Travel Days,” and a 30-day paid break every five years.

Is this job posting expired or no longer available?