Senior Machine Learning Engineer (LLM/RL)

Описание роли

На позиции Senior Machine Learning Engineer (LLM/RL) вы станете ключевым специалистом в разработке и масштабировании генеративных моделей искусственного интеллекта для молекулярного дизайна и открытия новых лекарств. Ваша основная ответственность — создание и оптимизация процессов fine-tuning больших языковых моделей с применением методов обучения с подкреплением (RLHF, PPO, DPO, GRPO) для генерации молекул и предсказания их свойств. Вы будете проектировать высокопроизводительные конвейеры обучения transformers в многосерверных кластерах с несколькими GPU-узлами и обеспечивать параллелизм данных, модели и pipeline, чтобы ускорять экспериментальную разработку.

Ваши ключевые задачи:
* Проектирование и реализация распределённых pipeline обучения на PyTorch, DeepSpeed и vLLM;
* Разработка комплексных reward functions и reward models для специфических биологических и химических целей;
* Настройка и контроль экспериментов с RLVR/RLHF, анализ ключевых метрик и их интерпретация;
* Оптимизация моделей и инфраструктуры для повышения throughput и снижения затрат на вычисления;
* Подготовка технических отчётов, презентаций и публикаций по результатам исследований.

Особенности вакансии

Работа в международной англоязычной команде позволяет обмениваться опытом с ведущими экспертами из разных стран. Компания предлагает:
* Полную занятость в формате офиса в Абу-Даби с организацией релокации;
* Современный технический стек: PyTorch, Hugging Face Transformers, TRL/RLlib, DeepSpeed, vLLM, Docker, AWS SageMaker;
* Участие в научно-исследовательских проектах мирового уровня и возможность выступать на профильных конференциях;
* Конкурентный уровень дохода и поддержку профессионального развития;
* Чёткие карьерные перспективы внутри быстрорастущей HealthTech/MedTech компании.

О компании

Компания представляет собой end-to-end фармацевтическую/биотехнологическую платформу на базе искусственного интеллекта. Миссия — ускорение открытия новых лекарств посредством объединения вычислительной химии, биоинформатики и клинических данных в единой инфраструктуре. Собственные GenAI-модели и проприетарная платформа позволяют уменьшить время и ресурсы на ранние стадии R&D, а интеграция RL-подходов обеспечивает создание высокоэффективных молекул.

Уникальность вакансии

Вы получите опыт работы на передовой исследовательской границе, объединяя трансформеры и reinforcement learning для молекулярного дизайна. Такой опыт крайне востребован в фармацевтической отрасли и станет мощным активом в вашем профессиональном портфолио.