Software Engineer, PyTorch - Deep Learning

Компания: NVIDIA

Должность: Инженер-программист, PyTorch - Глубокое Обучение

Местоположение: Возможность работы для глобальных талантов

Вы готовы к инновациям на стыке ИИ и программной инженерии? NVIDIA ищет квалифицированных инженеров-компиляторов глубокого обучения для создания передовых инструментов, повышающих возможности инженеров ИИ по всему миру. Воспользуйтесь возможностью повлиять на дизайн, разработку и развертывание ИИ в мировом масштабе, используя тысячи GPU. Присоединяйтесь к нашей отданной команде, сосредоточенной на постоянном достижении превосходства в программных решениях на ведущем оборудовании NVIDIA с PyTorch.

В качестве ключевого члена нашей команды ваша роль будет выходить за традиционные рамки, продвигая развитие распределенного обучения глубокого обучения через передовые технологии компиляторов. Ваши основные обязанности будут включать:

• Проведение глубокого анализа рабочих нагрузок глубокого обучения для выявления и устранения узких мест производительности, функциональных аномалий и неэффективности системы.
• Прямое связывание проблем производительности с дефектами компилятора или неиспользованным потенциалом оптимизации и формулирование надежных решений.
• Совместная работа над уточнением трансформаций программы или изобретением новых после всестороннего обзора производительности.
• Следить за последними разработками в области компиляторов глубокого обучения для предложения передовых решений, повышающих эффективность фреймворков глубокого обучения.
• Проведение тщательного тестирования и проверок для обеспечения оптимальной производительности и качества обучения моделей.

Приходите вооруженными:

• Бакалаврской, магистерской или докторской степенью в области информатики или смежных технических областей, либо сравнимым опытом.
• Знанием Python и знакомством с фреймворками машинного обучения, такими как PyTorch или JAX.
• Крепким пониманием концепций компиляторов, включая абстрактную интерпретацию, представления кода, такие как SSA форма и AST, генерацию кода и преобразование программ.
• Подтвержденным опытом разработки крупных программных проектов.
• Сильными способностями к устному и письменному общению