Senior ML Engineer / Research Engineer (NLP)

Just AI — лидер в области генеративного искусственного интеллекта и AI-решений для бизнеса. С 2016 года мы создаём передовые технологии на базе больших языковых моделей (LLM) и генеративного ИИ, интегрируя их в бизнес-процессы компаний различных отраслей.

Наша философия — быть первыми в инновациях. Мы разрабатываем решения, которые завтра станут стандартом, и гордимся тем, что наши технологические подходы получают признание на рынке.

Наша экспертиза включает разработку корпоративных GPT-решений, AI-ассистентов нового поколения, систем автоматизации на основе генеративных моделей и комплексных платформ для работы с неструктурированными данными.

Немного о цифрах в Just AI:

• 150+ крупных клиентов из финтеха, телекома, ритейла, страхования и других отраслей;
• 230+ тысяч пользователей уже выбрали нашу революционную платформу Jay Flow;
• 60+ компаний-партнёров;
• Команда из 250+ профессионалов, которые сегодня создают то, что казалось невозможным вчера.

Наша команда R&D ищет сильного ML-инженера с исследовательскими интересами и складом ума для разработки передовых NLP-решений. Вы будете ключевым участником команды, помогающей совершенствовать наши продукты Jay Knowledge Hub, Jay Guard, Agents Platform и других.

Ключевая задача: проведение end-to-end исследований и разработки в различных направлениях NLP, от прототипирования и проведения необходимых эмпирических экспериментов до передачи в высоконагруженный продакшн. Кроме того, до 10-15% времени может уходить на решение инженерных вопросов по Python сервисам продукта Caila.

Основные направления работы:

Мы инициируем исследования сами или решаем насущные вопросы в существующих продуктах. Вот некоторый пул тем, которые были у нас в разработке за последние полгода:

• Извлечение информации: задачи NER для динамической анонимизации текстов.
• Работа с текстом и изображениями: интеграция и улучшение компонентов OCR для обработки документов.
• Сложные семантические системы: совершенствование RAG-пайплайнов (retrieval-augmented generation), улучшение релевантности поиска и генерации ответов.
• Диалоговые AI-агенты: проектирование и реализация компонентов для виртуальных ассистентов, включая пайплайны ASR → LLM → TTS.
• LLM benchmarking: создание и сбор данных для проведения внутренних бенчмарков по приоритетным направлениям и задачам.
• Нагрузочное тестирование: разработка и внедрение инструментов для оценки производительности LLM моделей.

Задачи:

• Проведение полного цикла ML-исследований: анализ задачи, подбор методов, прототипирование, эксперименты, подготовка к продуктовому внедрению.
• Оптимизация инференса DL-моделей для высоконагруженной среды (работа с ONNX, квантизацией, распределением на несколько GPU).
• Улучшение архитектуры системы для автоматического обучения и дообучения моделей на данных пользователей.
• Поддержка и эксплуатация ML-моделей в продакшене: мониторинг, анализ проблем, реакция на инциденты, постоянная оптимизация производительности. Участие в командной работе: консультации, код-ревью, разработка автотестов.

Требования:

• Опыт end-to-end разработки и внедрения ML-решений на Python.
• Глубокие знания и практический опыт работы с PyTorch.
• Опыт решения промышленных задач в области NLP.
• Знание современных архитектур и подходов (трансформеры, техники для работы с LLM, RAG).
• Опыт работы с библиотеками и фреймворками для высокопроизводительного инференса: vLLM, ONNX Runtime, TensorRT.
• Понимание принципов построения высоконагруженных сервисов.

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

• Опыт работы с инструментами MLOps (MLFlow, DVC, ClearML и т.п.).
• Знакомство со стеком агентских систем или диалоговых AI (LangChain, LlamaIndex и аналоги).
• Опыт в смежных областях: Computer Vision (для OCR), аудиообработка (ASR/TTS).
• Публикации, участие в Kaggle или сильные open-source проекты в области NLP/ML.

Мы предлагаем:

• Интересные проекты, возможность быстрой имплементации результатов исследований в продукты;

• Возможности для профессионального роста и обмена опытом как внутри коллектива, так и в рамках внешнего сотрудничества с нашими партнерами, университетами и сообществами;

• Официальное оформление по ТК, ежегодный выбор бенефита (компенсация ДМС/спорта/обучения);

• Работа в гибридном формате из нашего офиса в Петербурге или удаленная работа;

• Just AI входит в группы IT-компаний i-Free, объединяющей 16 компаний со штатом около 1000 человек – обилие ивентов по обмену опытом, митапов, возможности для горизонтального развития в компаниях группы;

• Неповторимая корпоративная культура Just AI DNA, вечеринки, совместные кинопросмотры и многое другое…