Staff ML Data Scientist

Staff ML Data Scientist, LLM Recommender Systems

We’re looking for a top-tier ML Data Scientist to build, train, and deploy next-generation LLM-based recommender systems. You will work directly with a top notch science team, the AWS account team, and the AWS Applied Scientist leading the science. The role covers the full path from raw customer data to a deployed model. We are looking for a seasoned applied scientist with a multi-year record of production ML, not someone who recently moved over from analytics or data science generalism.

Why This Role Matters

At Robots & Pencils, we design AI systems for a human world. Our name says it all. Robots and pencils means engineering paired with creativity, because every system we ship has to work for real people in real workflows. Here, you shape how AI integrates into enterprise operations and how products create measurable impact for clients and the people they serve. We ship production-ready AI in 30 to 45 days. That pace demands people who take ownership, lead with craft, and care deeply about what they put their name on.

What You’ll Do

Turn structured and unstructured traveler and catalog data into narrative text documents and synthetic data, at a scale of tens of millions of documents

Make the feature and context engineering decisions: what information goes into each document and how to express it (e.g., market-relative price, tier, trip context)

Run large-scale distributed training on Amazon SageMaker AI, including HyperPod

Fine-tune models with LoRA/QLoRA adapters, supervised training, and reinforcement learning for ranking and reasoning

Own deep learning modeling, including architecture, ranking heads, and co-training and multi-task setups

Deploy low-latency, cost-efficient inference, including quantization, distillation, and profiling

Build evaluation sets, run ablations, catch data leakage, and write up findings clearly for Expedia and the AWS science team

Handle customer data safely, including personal-data removal, working inside the customer’s environment

Bring an AI-forward mindset to your daily work, using tools like Claude and Cursor to ship higher-quality work at pace

What You’ll Bring

Spark (PySpark) and the Hadoop ecosystem (Hive, HDFS) on corpora of tens of millions of documents

Hands-on Amazon SageMaker AI: processing, training, distributed training (HyperPod), and deployment

LLM fine-tuning (LoRA/QLoRA) and reinforcement learning (e.g., GRPO or PPO-style methods)

Strong deep learning fundamentals in PyTorch: architectures, training dynamics, co-training

Large-scale batch and production LLM inference (e.g., vLLM, Amazon Bedrock batch)

Data privacy and security on AWS: personal-data removal, IAM, KMS, Lake Formation

Clear written communication with customer and science stakeholders

Demonstrable usage of AI-forward tools such as Claude and Cursor

Helpful Extras and Unique Skills

AWS Trainium or Inferentia (Neuron SDK)

Recommender systems: ranking metrics (NDCG, MRR), sequential and generative recommenders

Embedding quantization or semantic IDs

Customer-facing or embedded consulting experience