Ведущий разработчик NLP (AI)

Локация: Москва и Краснодар — гибридный или офисный формат; другие города — удаленная работа

Компания

AI-команда развивает решения на основе машинного обучения и языковых моделей для продуктов «Магнита», включая «Магнит Метрику».

Что делать

• Подготавливать и версионировать датасеты для многоклассовой и мультилейбл-классификации пользовательских комментариев.
• Проводить EDA и контролировать качество данных, включая разметку, дисбаланс классов, дубликаты и утечки между train, validation и test.
• Разрабатывать пайплайны получения silver-разметки с использованием удаленных LLM через OpenAI-compatible API.
• Обучать и сравнивать NLP-модели и подходы к классификации: retrieval, cross-encoder, bi-encoder и teacher–student.
• Проводить воспроизводимые эксперименты, рассчитывать метрики, подбирать пороги и анализировать ошибки моделей.
• Поддерживать инфраструктуру экспериментов и inference: Docker, GPU, логирование, конфигурации, checkpoint/resume и документацию.

Требования

• Опыт коммерческой разработки NLP-решений и обучения моделей для классификации текстов.
• Уверенное владение Python и PyTorch.
• Практический опыт работы с Hugging Face Transformers, Datasets и scikit-learn.
• Понимание text classification, multiclass и multilabel classification, retrieval, cross-encoder и bi-encoder моделей.
• Опыт ML-экспериментов, работы с метриками и валидацией, подбора порогов и анализа ошибок.
• Опыт запуска и поддержки моделей в контейнерной среде; желательно знание Docker, GPU inference, vLLM, Triton или других inference-серверов.

Культура и преимущества

• Гибкое начало рабочего дня.
• ДМС со стоматологией и льготными условиями для членов семьи.
• Техника для работы.
• Оплата обучения на курсах и участия в конференциях.
• Корпоративная библиотека курсов, скидки PRIMEZONE и программы лояльности.