Machine Learning (ML) Engineer, Ranking & Retrieval, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home

WPP Media; United States; Remote; USD 120,000 - 160,000; Full time

Benefits offered by this job

Medical insurance

Dental insurance

Vision insurance

Savings plan options

Paid time off

Job summary

WPP Media Limited is seeking a Machine Learning Engineer to build and deploy ML models powering AI-native applications. You will work on classification, retrieval, ranking, and personalization, taking models from development to production.

This role is 100% remote in the United States and requires willingness to complete a 60-minute coding assessment. You will develop evaluation pipelines, manage data and metrics, and collaborate with engineering and product teams to deliver reliable ML-powered

Qualifications

• ML experience required.
• AI experience required.
• Strong foundations in ML and modern neural architectures.
• Hands-on production deployment of ML models.
• Experience in classification, ranking, retrieval, recommendation, or personalization.
• Understanding of evaluation, experimentation and metrics.
• Able to write production-quality code with real-world datasets.
• Balance model quality, latency, cost and reliability.

Responsibilities

• Build and improve ML models for classification, ranking, retrieval and personalization.
• Develop efficient classifiers and optimize accuracy, latency and cost.
• Improve search relevance and context retrieval using embeddings and ranking techniques.
• Build personalization models from user behavior while respecting privacy.
• Develop evaluation pipelines, datasets and metrics to measure model quality.
• Fine-tune and adapt existing models with sufficient data.
• Build and maintain training, inference and deployment pipelines.
• Debug production issues and monitor model performance to ship improvements.

Skills

ML experience

AI experience

Neural architectures

Production coding

Python

PyTorch/JAX

Ranking models

Vector search

Problem solving

Ownership

Tools

Python

PyTorch

JAX

Job description

Machine Learning (ML) Engineer, Ranking & Retrieval, Machine Learning (ML) and Artificial Intelligence (AI) Required, Work From Home

The Machine Learning (ML) Engineer will build and improve Machine Learning (ML) systems that power our AI-native applications. You will work on classification, retrieval, ranking, personalization and evaluation, taking ownership from model development to production deployment. This role is for engineers who enjoy solving practical Machine Learning (ML) problems, improving model quality and shipping reliable systems used by real users. Machine Learning (ML) and Artificial Intelligence (AI) experience are required. This position is 100% Remote.

MUST BE WILLING TO TAKE A 60 MINUTE CODING ASSESSMENT.

Machine Learning (ML) Engineer Responsibilities:

• Build and improve Machine Learning (ML) models for classification, ranking, retrieval and personalization.
• Develop efficient classifiers and optimize model accuracy, precision, latency and cost.
• Improve search relevance and context retrieval using embeddings, hybrid search and reranking.
• Build personalization models that learn from user behavior while respecting privacy.
• Develop evaluation pipelines, labeled datasets and metrics to measure model quality.
• Fine-tune and adapt existing models when supported by sufficient data.
• Build and maintain Machine Learning (ML) training, inference and deployment pipelines.
• Debug production issues, monitor model performance and ship improvements based on real-world feedback.

Machine Learning (ML) Engineer Outcomes:

• Machine Learning (ML) models in production meet expected accuracy, relevance, latency and reliability targets.
• Classification and ranking improvements deliver measurable gains in model quality.
• Evaluation pipelines identify regressions and support reliable model releases.
• Training, inference and data pipelines are robust, reproducible and maintainable.
• Model improvements are driven by real-world feedback and measurable results.
• Collaborates effectively with engineering and product teams to deliver reliable ML-powered features.

Machine Learning (ML) Engineer Qualifications:

• Machine Learning (ML) experience is required.
• Artificial Intelligence (AI) experience is required.
• Strong foundations in Machine Learning (ML) and modern neural architectures.
• Hands-on experience building and deploying Machine Learning (ML) models in production.
• Experience in classification, ranking, retrieval, recommendation systems or personalization.
• Understanding of model evaluation, experimentation and performance metrics.
• Comfortable writing production-quality code and working with real-world datasets.
• Able to balance model quality, latency, cost and reliability.
• Strong ownership, problem-solving skills
• Tech Stack: Embeddings, vector search and ranking models, Production Machine Learning (ML) systems and inference pipelines, Python, and PyTorch/JAX.

Benefits include medical insurance, Dental, Vision, Savings Plan Options, PTO, etc.

Additional Information

All your information will be kept confidential according to EEO guidelines.

Compensation: USD 120000 – USD 160000 – yearly