You will be a member of the Triton Accel team and part of the bigger industry-leading PyTorch AI framework organization. Triton is a breakthrough Python-based language and compiler for authoring performant kernels for AI Accelerators such as GPUs.Our team works on all aspects of Triton - compiler optimizations (architecture-independent and dependent), performance and debug tooling, authoring performant kernels, and more. Our emphasis is OSS, but we also work with internal product teams.Our internships are twelve (12) to twenty four (24) weeks long and we have various start dates throughout the year.Research Scientist Intern, Systems ML - Frameworks / Compilers / Kernels (PhD) Responsibilities:Work with Meta researchers to advance the state-of-the-art in ML Compiler space, with an emphasis on Triton/Triton-like specification languages (EXO, Pallas, Mosaic are examples of the latter).Publish these results at top conferences, and use this experience to create and socialize a research program for the ML Compiler space.Minimum Qualifications:
You will be a member of the Triton Accel team and part of the bigger industry-leading PyTorch AI framework organization. Triton is a breakthrough Python-based language and compiler for authoring performant kernels for AI Accelerators such as GPUs.Our team works on all aspects of Triton - compiler optimizations (architecture-independent and dependent), performance and debug tooling, authoring performant kernels, and more. Our emphasis is OSS, but we also work with internal product teams.Our internships are twelve (12) to twenty four (24) weeks long and we have various start dates throughout the year.Research Scientist Intern, Systems ML - Frameworks / Compilers / Kernels (PhD) Responsibilities:
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
