Machine Learning Engineer — San Jose, CA

We're looking for a Senior Data Scientist / ML Engineer to join a cutting-edge AI programme focused on building next-generation LLM-powered recommendation and personalisation systems.

You'll work hands-on with large-scale datasets, develop training pipelines, fine-tune foundation models, and deploy production-grade AI solutions in AWS.

Key Responsibilities

• Build training datasets from structured and unstructured data at scale.

• Develop and optimise LLM fine-tuning pipelines using LoRA, QLoRA and reinforcement learning techniques.

• Design and deliver distributed training workloads using Amazon SageMaker and HyperPod.

• Develop ranking, recommendation and reasoning models using PyTorch.

• Optimise inference performance through quantisation, distillation and profiling.

• Build evaluation frameworks, perform error analysis and communicate findings to key stakeholders.

• Ensure data privacy, governance and security standards are maintained.

Required Experience

• Strong Python, PyTorch and machine learning engineering background.

• Experience with PySpark, Spark, Hive and large-scale distributed data processing.

• Hands-on Amazon SageMaker experience, including model training and deployment.

• LLM fine-tuning, RAG, agentic AI and reinforcement learning.

• Production inference experience using technologies such as vLLM and Amazon Bedrock.

• Strong AWS security and data governance knowledge.

• Excellent communication and stakeholder management skills.

Nice to Have

• AWS Trainium, Inferentia and Neuron SDK.

• Recommender systems and ranking models (NDCG, MRR).

• Customer-facing or consulting experience.

Please note: Candidates must be based in the United States and able to spend a significant amount of time onsite in San Jose, CA. Unfortunately, visa sponsorship is not available for this position.