Our team makes PyTorch run faster and more resource-efficient without sacrificing the flexibility and ease of use of PyTorch.We are looking for a talented intern to join our team on an ambitious open-source, public initiative. The project is aimed at designing an LLM-based Triton and CUDA kernel generator through modern AI techniquesTeam 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-Contribute to the research direction of AI for GPU programming, https://gpu-mode.github.io/popcorn/Our team at Meta AI offers twelve (12) to sixteen (16) weeks long internships and we have various start dates throughout the year. To learn more about our research, visit https://ai.facebook.com.
Want more jobs like this?GetjobsinMenlo Park, CAdelivered to your inbox every week.
Want more jobs like this?
GetjobsinMenlo Park, CAdelivered to your inbox every week.
Get Jobs
