Lead ML Engineer для AI-компаньонов@ Joi AI

Что за роль

Joi AI развивает платформу AI-компаньонов и эмоционально интеллектуальные ИИ-модели. Ведущий ML-инженер будет оптимизировать LLM-инференс и технически руководить командами NLP и CV.

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

• Оптимизировать и масштабировать LLM-инференс с помощью SGLang, vLLM и TensorRT-LLM.
• Работать с моделями до 1T+ параметров и распределённой GPU-инфраструктурой.
• Обучать и дообучать LLM, развивать AI-агентов.
• Руководить ML-инженерами и определять техническую стратегию.

Требования

• Опыт оптимизации LLM-инференса в production.
• Знание распределённых вычислений, MoE, KV-cache, квантования и GPU-профилирования.
• Опыт обучения LLM и постобучения, включая RLHF/DPO.
• Владение PyTorch, Transformers и Python.
• Техническое лидерство и готовность самостоятельно писать код.
• Продвинутый английский или русский.

Будет плюсом

• CUDA/Triton, мультимодальные модели, вклад в open source или научные публикации.

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

• Полностью удалённую работу и полную занятость.
• Возможность получить долю в компании и вырасти до Head of ML.
• 28 дней отпуска и 7 wellness-дней.
• Медицинскую поддержку до $1 000 в год или страховку.
• Компенсацию рабочего места, обучения и конференций.