Staff Research Engineer (LLM Pre-Training)

Мы ищем Research Engineer, который сможет вносить вклад в обучение базовых моделей для задач программирования. Вы будете заниматься разработкой Large Language Models с нуля и их развертыванием в производственных средах, где они будут доступны конечным пользователям по всему миру.

• Работать со стейкхолдерами, преобразуя бизнес‑требования в технические спецификации.

• Обучать LLM с нуля на большом GPU‑кластере.

• Собирать и обрабатывать наборы данных для предобучения и дообучения.

• Поддерживать и улучшать существующие подсистемы.

• Опыт в проектировании, развертывании и поддержке производственных ML‑систем.

• Сильные теоретические знания в области NLP и подходов на основе трансформеров.

• Владение современными фреймворками глубокого обучения, такими как PyTorch, и распространёнными библиотеками для NLP.

• Опыт распределённого обучения моделей с миллиардами параметров.

• Внимание к деталям во всём, что вы делаете, и отличные коммуникативные навыки.

• Фреймворки инференса LLM, такие как vLLM, DeepSpeed, TensorRT.

• Техники выравнивания LLM, такие как RLHF/RLAIF.

• Инструменты и практики MLOps, включая CI/CD для ML.

• K8s и Kubeflow.

• Научные публикации в области NLP.

Как мы развиваем JetBrains AI:

• Кластер из сотен NVIDIA GPU в качестве инфраструктуры для обучения.

• Git для управления исходным кодом.

• Python, PyTorch и HuggingFace как ML‑стек.

• Kubeflow и Weights & Biases для отслеживания экспериментов.

• TeamCity как система автоматизации CI.

#LI-KP1

Мы являемся работодателем, предоставляющим равные возможности

Мы знаем, что отличные идеи могут появиться у любого, где бы он ни находился. Поэтому мы делаем всё возможное, чтобы создать открытое и инклюзивное рабочее место – место, где каждый будет принят независимо от своего происхождения, идентичности, религии, возраста, потребностей в доступности или ориентации.

Мы обрабатываем данные, предоставленные в вашей заявке, в соответствии с Политикой конфиденциальности при наборе персонала.