Senior ML Engineer - LLMs for Multi-Step Coding Agents

HireHi; United States; Remote; USD 180,000 - 240,000; Full time

Job summary

JetBrains ищет инженера-исследователя для разработки и обучения многошаговых кодирующих агентов в IDE. Вы будете строить SFT и RL-пайплайны, адаптировать LLM под задачи планирования инструментов и взаимодействий внутри среды JetBrains IDE.

Требуется глубокий опыт обучения LLM, работа с PyTorch и стеков Megatron/NeMo/Verl, умение руководить проектами от идеи до выпуска, нацеленность на продукт и умение работать с распределёнными вычислениями на GPU.

Qualifications

• Обширный практический опыт обучения LLM в исследовательской или продакшн-среде.
• Глубокая экспертиза современных фреймворков DL (PyTorch) и стеков обучения LLM (Megatron, NeMo, Verl).
• Сильное понимание архитектур LLM, токенизации, пайплайнов данных и распределённых вычислений.
• Способность руководить проектами от идеи до реализации и оценки.
• Ориентация на продукт и умение переводить потребности разработчиков в моделирование и оценку.
• Не менее 3 лет опыта программирования на Python в ML-проектах.

Responsibilities

• Разработка, внедрение и сопровождение SFT и RL-пайплайнов для многошаговых кодирующих агентов.
• Обучение и адаптация LLM под сценарии использования внутри IDE JetBrains.
• Создание сред для оценки и симуляции агентов; измерение и сравнение по реальным задачам.
• Проектирование фреймворков оценки и метрик, анализ логов и тренировочных данных.
• Анализ результатов обучения и предложение улучшений архитектур и рецептов обучения.
• Работа на крупном инфраструктурном стеке: распределённое обучение на GPU-кластерах и MapReduce-подходах.
• Сотрудничество с исследованиями, продуктом и инфраструктурной командой для реализации идей в готовые фичи.

Tools

Megatron

NeMo

Verl

Python

Distributed training

Experiment tracking

Job description

JetBrains ищет инженера-исследователя для разработки и обучения многошаговых кодирующих агентов в IDE. Вы будете строить SFT и RL-пайплайны, адаптировать LLM под задачи планирования инструментов и взаимодействий внутри среды JetBrains IDE.

Требуется глубокий опыт обучения LLM, работа с PyTorch и стеков Megatron/NeMo/Verl, умение руководить проектами от идеи до выпуска, нацеленность на продукт и умение работать с распределёнными вычислениями на GPU.