Software Engineer, Systems ML - PyTorch Compiler / ML Framework / Performance

Присоединяйтесь к команде компилятора PyTorch в Meta

Обзор:

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

Основные обязанности:

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

Минимальные требования:

• Бакалавр или магистр в области компьютерных наук или близкой по тематике специальности.
• Практический опыт в области компиляторов, систем машинного обучения, высокопроизводительных вычислений и оптимизации для GPU.
• Владение программированием на Python или CUDA.
• Обучение или полученная степень бакалавра в области компьютерных наук или смежной технической специальности.

Предпочтительные квалификации:

• Доктор философии в области компьютерных наук, прикладной математики или смежных дисциплин.
• Опыт работы с технологиями PT2, Triton, MLIR или прямой вклад в PyTorch.
• Глубокие знания в области производительности GPU и разработки высокопроизводительных ядер CUDA.
• Подтвержденный опыт исследовательской и инженерной работы, подтвержденный публикациями, патентами или участием в соответствующих конкурсах.

О Meta:

Meta продолжает переопределять ландшафт связи и социального взаимодействия с помощью инноваци