Fulltime :: Senior Data Scientist / ML Engineer :: San Jose, CA (Onsite)
Quick overview
Job description
Summary Works directly with Client and its data to build, train and deploy next-generation LLM-based recommender systems. Works closely with the client's 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.
Responsibilities:
- Data to training sets: turn unstructured and structured traveler and catalog data into narrative text documents and synthetic data, at a scale of tens of millions of documents.
- Feature and context engineering: decide what information goes into each document and how to express it, such as market-relative price, tier and trip context.
- Large-scale distributed training on Amazon SageMaker AI, including HyperPod.
- Fine-tuning: LoRA/QLoRA adapters, supervised training, and reinforcement learning for ranking and reasoning.
- Deep learning modeling: model architecture, ranking heads, co-training and multi-task setups.
- Inference and deployment: low-latency, cost-efficient serving, including quantization, distillation and profiling.
- Evaluation and error analysis: build evaluation sets, run ablations and catch data leakage. Write up findings clearly for the customer and the AWS science team.
- Data privacy: handle customer data safely, including removing per
Required skills
- Spark (PySpark) and the Hadoop ecosystem (Hive, HDFS) on corpora of tens of millions of documents.
- Amazon SageMaker AI: processing, training, distributed training (HyperPod), deployment.
- LLM fine-tuning (LoRA/QLoRA) and reinforcement learning (e.g. GRPO/PPO-style methods).
- Strong deep learning fundamentals in PyTorch: architectures, training dynamics, co-training.
- Large-scale batch LLM inference and production inference (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.
Preferred skills
- AWS Trainium / Inferentia (Neuron SDK).
- Recommender systems: ranking metrics (NDCG, MRR), sequential and generative recommenders.
- Embedding quantization or semantic IDs.
Experience in a customer-facing or embedded role
Similar jobs
- ST

Senior Machine Learning Engineer–Field EngagementRecently posted Auto Apply
Striveworks · Washington · Hybrid 🇺🇸
OnboardingETLCompliance+3Technology$160k - $200k/yr - TA

Perception Engineer, Missile DefenseRecently posted Auto Apply
True Anomaly · Denver · Hybrid 🇺🇸
C++PythonEngineeringSalary not listed - OP

Machine Learning Performance EngineerRecently posted Auto Apply
Optiver · New York · Hybrid 🇺🇸
CUDADeep LearningC+++5Technology$200k/yr - HA
Senior Machine Learning EngineerRecently posted Auto Apply
Hum Ai · San Francisco · Hybrid 🇺🇸
FirebaseAWSPyTorch+3TechnologySalary not listed - GL

Machine Learning Engineer, Enterprise BrainRecently posted Auto Apply
Glean · Mountain View · Hybrid 🇺🇸
GitGDPRLLM+5Technology$200k - $280k/yr - RD

ML EngineerRecently posted
Randstad Digital · New York, NY · Hybrid 🇺🇸
NLPDeep LearningLLM+4Technology$75 - $85/hr - VG

AWS AI/ML SME (100% Remote) USA Citizens only
VentureSoft Global · United States · Remote 🇺🇸
AWSMLOpsReact+1TechnologySalary not listed - CL

Coding RL EngineerRecently posted Auto Apply
Clera · San Francisco · On-site 🇺🇸
DockerRustJava+2EngineeringSalary not listed - CH

Senior Machine Learning EngineerRecently posted Auto Apply
Checkr · Denver · Hybrid 🇺🇸
RecruitingRubyApache Spark+12TechnologySalary not listed - BU

Reinforcement Learning EngineerRecently posted Auto Apply
Bugcrowd · Remote - US · Remote 🇺🇸
DockerRustGit+7Engineering$176.4k - $242.6k/yr - BO

Machine Learning Engineer III, ShieldRecently posted Auto Apply
Box · Redwood City · Hybrid 🇺🇸
GCPScalaApache Spark+11Technology$175.5k - $219.5k/yr - AX

Senior Machine Learning Engineer IRecently posted Auto Apply
Axon · Seattle · Hybrid 🇺🇸
GCPAWSLLM+6Technology$150.8k - $241.2k/yr