Principal Machine Learning Engineer, Presentation and Visual Optimization

Paramount

Principal Machine Learning Engineer at Paramount focusing on artwork personalization and attention optimization for streaming services. Leading experimentation and developing machine learning strategies for visual content presentation.

Posted 5/1/2026full-timeRemote • New York • 🇺🇸 United StatesMid-LevelSenior💰 $234,000 - $250,000 per yearWebsite

Tech Stack

Tools & technologies

PythonPyTorchSparkTensorflow

About the role

Key responsibilities & impact

Define the technical roadmap for visual personalization, bridging the gap between ML science and UI/UX design
Architect and deploy multi-armed bandit (MAB) systems to dynamically select the best creative assets for show tiles and hero marquees
Develop models to personalize carousel design types (e.g., square vs. poster), presentation styles, and even the ordering of attributes shown to the user
Utilize NLP and LLMs to experiment with and optimize carousel titles and content descriptions based on user preferences and trending topics
Design "always-on" experimentation pipelines that can shift traffic based on model confidence and reward signals (CTR, Playback Start)
Partner with Content Engineering to leverage visual embeddings and computer vision signals to understand why certain artwork performs better than others

Requirements

What you’ll need

6-8+ years of experience in machine learning engineering or applied science
Deep hands-on experience with Multi-Armed Bandits, Contextual Bandits, Thompson Sampling, or Upper Confidence Bound (UCB) algorithms
Proven track record of designing high-velocity A/B testing or online learning systems
Proficiency in Python, PyTorch/TensorFlow , and big-data processing (Spark/Databricks)
Experience leading a technical pod or team, with a focus on translating product/design needs into engineering requirements
Experience in Visual Personalization (artwork, thumbnails, or creative optimization)
Background in Computer Vision (OCR, aesthetic scoring, or image feature extraction)
Knowledge of Reinforcement Learning (RL) for sequential layout optimization
Familiarity with LLMs/Generative AI for automated copy generation and title testing

Benefits

Comp & perks

medical
dental
vision
401(k) plan
life insurance coverage
disability benefits
tuition assistance program
PTO

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

machine learning engineeringMulti-Armed BanditsContextual BanditsThompson SamplingUpper Confidence Bound (UCB)A/B testingPythonPyTorchTensorFlowbig-data processing

Soft Skills

leadershipteam collaborationcommunicationtranslating product needsengineering requirements