ML engineer

Wildberries; United States; Remote; USD 140,000 - 190,000; Full time

Benefits offered by this job

ДМС спорт

Спортивные программы

Психологические консультации

Клубы сотрудников

Гибкий график

Офисы рядом с метро

Job summary

Wildberries ищет Middle+ / Senior ML Engineer в команду Поиска. Ваша задача улучшать пайплайны обработки текста в запросах и карточках товаров, развивать онлайн-инференс и строить модели для выделения текстовых синонимов и ранжирования. Требуется опыт работы с трансформерами, Spark/Airflow и векторным поиском.

Работа в международной среде, фокус на продуктивности и latency. Кандидат должен иметь минимум 3 года опыта ML/NLP, владение Python (Polars, PyTorch) и опытом внедрения в продакшн.

Qualifications

• Опыт работы ML/NLP инженером от 3 лет
• Уверенное владение Python и ML/DS-стеком (Polars, PyTorch и т.д.)
• Опыт работы со Spark и Airflow (ETL пайплайны)
• Опыт разработки и внедрения трансформерных моделей (HuggingFace)
• Опыт работы с табличными моделями (LightGBM, CatBoost, XGBoost)
• Опыт кластеризации (IVF/ANN-поиск)
• Понимание задач NLP: нормализация, опечатки, синонимия, классификация

Responsibilities

• Улучшать текущие и развивать новые пайплайны обработки текста в запросах и карточках товаров (опечатки, синонимизация)
• Развивать направление онлайн-инференса в продуктах обработки текста
• Разрабатывать пайплайн автоматического выделения текстовых синонимов для ретривинга и ранжирования
• Разрабатывать модель классификации запросов (текстовые, категорийно-атрибутивные, брендовые, нетекстовые)
• Исследовать методы кластеризации для ускорения и повышения качества векторного поиска

Skills

Python

ML/NLP engineering

Spark

Airflow

PyTorch

HuggingFace Transformers

LightGBM

CatBoost

XGBoost

ANN / clustering (IVF)

FAISS / HNSW

NLP text normalization

Tools

Spark

Airflow

HuggingFace Transformers

LightGBM

XGBoost

FAISS

Job description

Описание: Объединённая компания Wildberries и Russ — это международная технологическая компания, образованная в результате слияния двух лидеров рынка — IT-компании Wildberries и оператора наружной рекламы Russ. Мы ищем Middle+ / Senior ML Engineer в команду Поиска для развития направления Спеллчека пользовательских запросов и карточок товаров. При этом мы занимаемся широким спектром задач, связанных с обработкой текста в движке Поиска.

Задачи

• Улучшать текущие и развивать новые пайплайны обработки текста в запросах и карточках товаров (исправление опечаток, синонимизация)
• Развивать направление онлайн-инференса в продуктах обработки текста
• Разрабатывать пайплайн автоматического выделения текстовых синонимов для применения в ретривинге и ранжировании
• Разрабатывать модель классификации запросов (текстовые, категорийно-атрибутивные, брендовые, нетекстовые)
• Исследовать методы кластеризации (IVF и др.) для ускорения и повышения качества векторного поиска

Требования

• Опыт работы в роли ML / NLP Engineer от 3 лет
• Уверенное владение Python и стандартным ML/DS-стеком (polars, pytorch и т.д.)
• Опыт работы со Spark и Airflow (построение и поддержка пайплайнов обработки данных, ETL)
• Опыт разработки и внедрения трансформерных моделей (обучение, файнтюнинг, инференс — HuggingFace Transformers и т.п.)
• Опыт работы с табличными моделями (LightGBM, CatBoost, XGBoost) — построение ранжирующих/классификационных моделей
• Опыт применения методов кластеризации (например, IVF и другие подходы для векторных индексов/ANN-поиска)
• Понимание задач NLP: нормализация текста, работа с опечатками, синонимия, классификация текстов
• Опыт вывода моделей в продакшн, понимание требований к онлайн-инференсу (латентность, нагрузка)
• Умение работать с большими объёмами данных, оптимизировать пайплайны обработки
• Будет плюсом: Опыт работы с поисковыми системами (retrieval, ranking, семантический поиск) и векторными базами данных / библиотеками ANN-поиска (FAISS, HNSW и т.п.)

Условия

• Обучение и развитие: языковые клубы, собственный корпоративный университет, программы развития управленческих навыков и многое другое
• Благополучие сотрудников: корпоративный пакет ДМС со стоматологией, корпоративный спорт, консультации психолога и дополнительные возможности аккредитованной IT-компании
• Множество сообществ: клуб спикеров, футбола, йоги, шахмат и т.д.
• Забота о семьях: создаем условия, в которых легко сочетать карьеру и заботу о близких – от гибкого подхода до масштабных проектов для детей сотрудников
• Скидки и партнерские программы: на обучение, страхование, покупки и многое другое
• Комфортная рабочая среда: бесплатное питание в офисе, современные офисы рядом с метро, корпоративная техника и портал для сотрудников