Senior ML Engineer (LLM/RL)

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

На позиции Senior ML Engineer (LLM/RL) вы будете ключевым специалистом, отвечающим за разработку и масштабирование конвейеров fine-tuning больших языковых моделей с методами Reinforcement Learning. Ваша задача — адаптация GenAI-моделей для задач молекулярного дизайна и предсказания свойств соединений, что напрямую влияет на ускорение этапов открытия лекарств. Вы будете планировать эксперименты, отслеживать ключевые метрики производительности и качества моделей, а также оформлять результаты в виде технических отчетов, презентаций и публикаций.

Обязанности

* Разрабатывать и масштабировать процессы тонкой настройки (fine-tuning) LLM с обучением с подкреплением (RLHF, RLVR, PPO, DPO, RLOO);
* Строить и оптимизировать распределенные конвейеры тренировки трансформеров в средах multi-GPU и multi-node с использованием DeepSpeed и vLLM;
* Проектировать функции и модели вознаграждения (reward functions и reward models) для специфических задач молекулярного дизайна;
* Интегрировать и разворачивать модели на платформах Docker и AWS SageMaker с настройкой CI/CD;
* Руководить экспериментами, оценивать результаты, улучшать метрики качества и производительности моделей;
* Сотрудничать с междисциплинарной командой биологов, химиков и клиницистов, используя английский язык для общения и документации.

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

Эта позиция позволит вам стать частью международной англоязычной команды в офисе Абу-Даби с полным релокационным пакетом. Вы получите доступ к собственным платформам и данным для разработки AI-решений в фармацевтической отрасли, возможность публиковать результаты исследований и посещать профильные конференции. Гибкая архитектура конвейеров и свобода выбора инструментов: от Hugging Face Transformers до RLlib и DeepSpeed.

О компании

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