Middle ML Engineer

В команду рекомендаций ищем сильного ML Engineer от уровня Middle.

Wildberries – это 9 млн заказов и 1 млрд поисковых запросов в сутки. Команда рекомендаций помогает найти пользователю интересные ему товары в огромном ассортименте Wildberries. Она занимается персонализированным подбором товаров для пользователя исходя из его интересов, подбором похожих и сопутствующих товаров.

В отделе рекомендаций есть 3 проекта над которым предстоит работать:

• Выделение главного объекта на изображении. Это требуется для улучшения качества алгоритмов поиска похожих товаров. Сейчас модель работает на одежде, а хотелось бы, чтобы работало на всех категория Wildberries. Рассматриваем подходы по zero\few-shot detection (по типу GroudingDINO), а также с помощью стандартных детекторов (по типу YOLO);
• Модель предсказания атрибутов товаров, таких как категория, сезон, цвет и множество других. Такая модель может пригодиться для проектов, начиная от помощи в заполнении карточки товара, и заканчивая обогащением данных для поискового индекса;
• Реализация алгоритмов персональных рекомендаций на основе мультимодальных эмбеддингов товаров.

Требования:

• Понимание, как работает классический ML, в частности рекомендательные системы
• Опыт работы как с языковыми, так и с картиночными моделями нейронных сетей
• Знание базовых алгоритмов и структур данных
• Опыт выкатки моделей в production

Будет преимуществом:

• Обучение и дообучение нейросетей на PyTorch
• Опыт использования инструментов для совместной работы над моделями и их постоянной доработки (MLflow/ClearML, DVC)
• Опыт работы с большими данными, применения распределённых вычислений
• Умение делать inference для моделей, проектировать, разрабатывать и разворачивать собственные сервисы

Наш стек:

• Python, Go, Mlflow, ClearML, Prometeus, Grafana, Docker, PyTorch, Airflow, Postgres, Clickhouse, K8s, FastAPI, Gitlab, PyTorch, Tensorflow, ONNX

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

• Оформление по ТК, ГПХ, ИП
• Гибкий график, с возможностью полной удаленки из любой точки мира
• Возможность брать 3 day-off в любой день
• Необходимое мощное железо и ПО
• Ежегодная 40%-ая скидка на покупку ноутбука или мобильного телефона
• Бесплатное 3х разовое питание в виде шведского стола и мороженое в каждом офисе
• Скидки на обучающие курсы и фитнес-программы
• Карьерный и профессиональный рост
• ДМС после 3-х месяцев испытательного срока: телемедицина, очные приемы, ежегодный чек-ап для сотрудника, а также онлайн-консультации для родственников