Senior Software Engineer, JAX

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

Senior-инженер в команду NVIDIA, развивающую JAX — фреймворк для численных вычислений и глубокого обучения. Задача — вносить вклад в JAX, реализовывать ключевые компоненты и оптимизировать производительность на продуктах NVIDIA.

Задачи

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

Требования

• Степень бакалавра по Computer Science, Computer Engineering или смежной специальности (или эквивалентный опыт)
• 6+ лет релевантного опыта
• Программирование на C/C++ и Python
• Опыт работы с фреймворками машинного обучения и их внутренним устройством (например, PyTorch, TensorFlow, scikit-learn)
• Подтверждённый опыт разработки клиентских решений с балансировкой между запросами на фичи и багами
• Техническая база в архитектурах CPU и GPU, численных библиотеках, модульном проектировании ПО
• Высокая мотивация и отличные навыки устной и письменной коммуникации
• Умение работать с кросс-функциональными командами, принципалами и архитекторами, эффективно координироваться между подразделениями и регионами

Условия

• Базовая зарплата 184 000–287 500 USD для уровня 4 и 224 000–356 500 USD для уровня 5
• Право на equity и льготы
• Приём заявок как минимум до 15 октября 2026 года

Навыки

• JAX
• Python
• C
• C++
• NumPy
• PyTorch
• TensorFlow
• scikit-learn
• CPU
• GPU
• Autograd
• DSL compilers
• distributed systems