Приєднуйтесь до нашої команди компілятора PyTorch в Мета

Огляд:

Команда компілятора PyTorch компанії Meta прагне підвищити продуктивність PyTorch, зосереджуючись на ефективності та швидкості, не поступаючись його відомою гнучкістю. Наша революційна технологія PT2 з моменту свого створення у березні 2023 року встановила новий стандарт у ландшафті фреймворків машинного навчання, посуваючи PyTorch до позиції провідного ML компілятора в індустрії. Ми шукаємо інженера-програміста, який долучиться до роботи над компілятором PT2 та просуватиме межі технологій машинного навчання.

Основні обов’язки:

• Розробка та вдосконалення компонентів компілятора PT2: TorchDynamo, TorchInductor та PyTorch Core.
• Сприяння покращенню продуктивності PyTorch, пропонуючи систематичні рішення, що приносять користь широкій спільноті PyTorch.
• Дослідження та оптимізація генеративних моделей ШІ, зосередження на етапах попереднього навчання, доопрацювання і виведення.
• Дослідження синергії між компілятором PyTorch і розподіленими системами PyTorch.
• Активна співпраця з користувачами PyTorch для сприяння впровадженню та інноваційному використанню технологій PT2.

Мінімальні вимоги:

• Бакалавр/магістр в області комп'ютерних наук або суміжних дисциплін.
• Практичний досвід роботи з компіляторами, системами МЛ, обчисленнями високої продуктивності та оптимізацією для GPU.
• Володіння програмуванням на Python або CUDA.
• Завершення або отримання ступеня бакалавра в галузі комп'ютерних наук або спорідненої технічної спеціальності.

Бажані кваліфікації:

• Ph.D. в галузі комп'ютерних наук, прикладної математики або суміжних дисципл