Research Scientist Intern, PyTorch Compiler (PhD)
MetaCalifornia, United States • Seattle, Washington, United States • New York, New York, United States
Posted March 20, 2026
91800 - 145608 USD
Internship
About the Company
Meta builds technologies that help people connect, find communities, and grow businesses.
Job Description
Our team makes PyTorch run faster and more resource-efficient without sacrificing the flexibility and ease of use of PyTorch. We offer internships from 12 to 24 weeks and have various start dates throughout the year.
Requirements
- 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
- 12 to 24 weeks long internship
- Variety of start dates throughout the year
- Compensation: $7,650/month to $12,134/month