Research Scientist Intern, PyTorch Compiler

Our team makes PyTorch run faster and more resource-efficient without sacrificing the flexibility and ease of use of PyTorch. Team scope: - Advance PyTorch 2.0 technologies that bring torch.compile() to the heart of PyTorch - Advance PyTorch Distributed through torch.compile() - Improve PyTorch out-of-the-box performance on GPU, CPU, accelerators - Vertical performance optimization for models for training and inference Our team at Meta AI offers twelve (12) to sixteen (24) weeks long internships and we have various start dates throughout the year.