ЯндексML-разработчик в команду генерации изображений Alice AI ART
Команда занимается end-to-end разработкой и выводом в прод моделей генерации и редактирования изображений: от pretrain и SFT до RL и дистилляции. Роль важна для качества визуального контента в продукте и снижения latency и стоимости инференса.
ГрейдMiddle/Senior
Стек
PythonPyTorchTensorRTDockerPyTorch ProfilerNsight SystemsFSDPZeRO
Задачи
- •Развитие единой модели для text-to-image, instruction-based editing и multi-image (pretrain, SFT, RL, дистилляция)
- •Улучшение управляемости и соответствия промптов через кондишнинг и текстовые энкодеры
- •Исследование и реализация методов дистилляции для сокращения шагов генерации и latency
- •Профилирование и оптимизация работы на GPU, интеграция ускорений в прод-окружение
- •Формирование состава данных и масштабирование распределённого обучения
Обязательно
- •Писать понятный воспроизводимый код и доводить идеи до результата
- •Понимание или готовность быстро изучить диффузионные модели
- •Умение аккуратно ставить эксперименты и оценивать реальные улучшения
- •Опыт или желание работать с обучением больших моделей на нескольких GPU
- •Знание практик воспроизводимости и чекпойнтинга экспериментов
Будет плюсом
- •Опыт работы с диффузионными моделями, дистилляцией или image editing
- •Опыт обучения генеративных моделей (диффузия, GAN) в distributed-сетапах
- •Знание архитектур и трюков обучения LLM/VLM
- •Умение профилировать и ускорять код (PyTorch Profiler, Nsight, TensorRT) и работа с Docker
