LLM Engineer - Train & Deploy Large NLP Models

HireHi; United States; Remote; USD 120,000 - 210,000; Full time

Job summary

JetBrains ищет ML-инженера, который будет разрабатывать и поддерживать продакшн ML-системы, обучать и настраивать крупные модели на GPU-кластере, собирать данные и улучшать подсистемы AI.

Требуется сильная теория NLP и трансформеров, опыт работы с PyTorch и NLP-библиотеками, умение планировать проекты и принимать решения независимо, а также способность адаптироваться к требованиям заказчиков и постоянно учиться.

Qualifications

• Обширный опыт проектирования, развёртывания и поддержки продукционных ML-систем
• Сильная теоретическая база в NLP и трансформерах
• Умение работать с современными фреймворками DL, такими как PyTorch, и NLP-библиотеками
• Опыт распределённой тренировки миллиардных параметров
• Внимательность к деталям и хорошие коммуникативные навыки
• Способность планировать проекты и принимать решения независимо, с привлечением коллег по мере необходимости
• Умение распознавать потребности заказчиков и расставлять приоритеты
• Старт с простых решений и последующее наращивание сложности
• Готовность нести ответственность за подсистему
• Страсть к обучению и следованию за новинками в области LLM
• Будет плюсом: опыт в инференс-фреймворках (vLLM, DeepSpeed, TensorRT); техники согласования LLM; MLOps; Kubernetes/Kubeflow; публикации в NLP

Responsibilities

• Работа с заинтересованными сторонами над техническими спецификациями на основе бизнес-требований
• Обучение и настройка LLM на крупном GPU-кластере
• Сбор и обработка датасетов до/после обучения
• Поддержка и улучшение существующих подсистем ML

Skills

Designing ML systems

NLP theory

Transformer models

Project planning

Independent decision making

Customer needs prioritization

Communication skills

Learning mindset

Tools

PyTorch

NLP libraries

Job description

JetBrains ищет ML-инженера, который будет разрабатывать и поддерживать продакшн ML-системы, обучать и настраивать крупные модели на GPU-кластере, собирать данные и улучшать подсистемы AI.

Требуется сильная теория NLP и трансформеров, опыт работы с PyTorch и NLP-библиотеками, умение планировать проекты и принимать решения независимо, а также способность адаптироваться к требованиям заказчиков и постоянно учиться.