ML Research — Diffusion Models и Reinforcement Learning@ Higgsfield AI

О позиции

Higgsfield AI — одна из самых быстрорастущих GenAI-компаний в мире. Мы строим новую research-лабораторию, полностью сфокусированную на Diffusion Models, Multimodal AI и Reinforcement Learning. Начинаем с бюджетом $100M и дальше масштабируемся до $1B. Вход в LLM frontier стоит сотни миллиардов долларов, тогда как в диффузионных моделях индустрия только начинается.

Чем вы будете заниматься

• Исследовать архитектуры Diffusion Models, Multimodal AI и MoE.
• Работать с Reinforcement Learning для генеративных моделей.
• Проектировать отказоустойчивые training pipelines с максимальной утилизацией GPU на больших кластерах.
• Тренировать модели для фильтрации и оценки мультимодальных данных для обучения.
• Ускорять инференс и улучшать модели с помощью дистилляции, alignment и квантизации.

Требования

• Работаете в области Diffusion Models, Multimodal AI или Reinforcement Learning.
• Уверенно владеете PyTorch, включая torch.distributed.
• Работали с распределённым обучением, большими моделями или современным ML-стеком (DeepSpeed, FSDP и др.).
• Любите исследования и хотите превращать идеи в работающие модели.

Будет большим плюсом

• Публикации на NeurIPS, ICML, ICLR.
• Обучение или выпуск из ШАД, ФКН ВШЭ (ПМИ) или других сильных технических школ.
• Участие в олимпиадах или сильный исследовательский и инженерный бэкграунд.

Что мы предлагаем

• Офис в Казахстане, Алматы.
• Полная поддержка переезда в США.
• Вилки junior-senior 100-350k USD/year.
• Не обязательно соответствовать каждому пункту. Если у вас сильная математическая база, исследовательское мышление и желание строить новое поколение ML-моделей — будем рады познакомиться.