Lead ML Engineer
Joi AI — платформа AI-компаньонов, развивающая эмоционально интеллектуальные ИИ-модели. Ищем ведущего ML-инженера для оптимизации LLM и технического руководства командами NLP и CV.
Задачи
•Оптимизация и масштабирование LLM-инференса (SGLang, vLLM, TensorRT-LLM).
•Работа с моделями до 1T+ параметров и распределённой GPU-инфраструктурой.
•Обучение и дообучение LLM, развитие AI-агентов.
•Техническое руководство ML-инженерами и определение технической стратегии.
Требования
•Production-опыт оптимизации LLM-инференса.
•Знание распределённых вычислений, MoE, KV-cache, квантования и GPU-профилирования.
•Опыт обучения LLM и постобучения (RLHF/DPO).
•PyTorch, Transformers, Python.
•Техническое лидерство и готовность писать код самостоятельно.
•Продвинутый английский или русский.
Будет плюсом
•CUDA/Triton, мультимодальные модели, open-source вклад, научные публикации.
Условия
•Полностью удалённая работа, полная занятость.
•Возможность получить долю в компании и вырасти до Head of ML.
•28 дней отпуска + 7 wellness-дней.
•Медицинская поддержка до $1 000 в год или страховка.
•Компенсация рабочего места, обучения и конференций.
Отбор
1.Скрининг с рекрутером и 5 техническими вопросами.
2.Техническое интервью на 90 минут: live coding и ML System Design.
3.Финальное интервью с командой.