Senior Software Engineer, GNN

Автоматический пересказ вакансии на русском. Полный текст на языке оригинала и форма отклика — на сайте работодателя.

Senior Software Engineer в команду NVIDIA, создающую крупномасштабные ML-решения: Graph Neural Networks, табличные foundation-модели и ансамблевые модели. Роль — ускорение PyTorch-фреймворков и обучение/инференс моделей на GPU-инфраструктуре.

Задачи

• Разрабатывать ускоренные решения на PyTorch для крупномасштабных ML-моделей (GNN, TFM, ансамбли) с фокусом на эффективное обучение и инференс на GPU
• Поддерживать CUDA-X Libraries и интеграции для PyTorch-воркфлоу
• Совместно с разработчиками, продакт-менеджерами и учёными создавать GNN-модели и GPU-ускоренные реализации
• Помогать клиентам внедрять аппаратные и программные решения NVIDIA, собирать технические требования от клиентов и Solutions Architects
• Обеспечивать техническое лидерство и менторство инженеров команды
• Находить возможности улучшения кодовой базы и снижения затрат на поддержку через реархитектуру
• Применять agentic coding-инструменты для поиска и исправления багов, реализации фич и рефакторинга
• Решать сложные технические задачи и координировать работу между командами

Требования

• Bachelor’s degree или эквивалентный опыт плюс 5+ лет опыта в крупномасштабном ML, deep learning и data science; либо Master’s/PhD плюс 3+ года опыта
• 3+ года опыта с PyTorch
• 2+ года опыта обучения enterprise-масштабных ML-моделей на распределённой инфраструктуре
• 2+ года опыта проектирования и эксплуатации эффективных training/inference-воркфлоу на GPU, включая профилирование, масштабирование, оркестрацию и утилизацию ресурсов
• Отличные навыки программирования на C++ и проектирования ПО
• Опыт разработки, отладки и оптимизации высокопроизводительных приложений, желательно с GPU-ускорением на CUDA
• Сильные навыки коллаборации, коммуникации и документирования

Условия

• Базовая зарплата 152 000–241 500 USD для Level 3 и 184 000–287 500 USD для Level 4
• Equity и льготы
• Инклюзивная среда и равные возможности трудоустройства
• Заявки принимаются как минимум до 19 сентября 2026

Навыки

• PyTorch
• C++
• CUDA
• Graph Neural Networks
• GPU
• PyTorch Geometric
• Snowflake
• Databricks
• FAISS
• Milvus
• distributed training
• inference optimization