Senior Machine Learning Engineer – LLMs
Core
Train domain-specific large language models and provide technical leadership for the ML team.
Role type
Senior Machine Learning Engineer (LLMs)
Builds
Production LLMs and training infrastructure for Prosus's AI-driven ecosystem
Domain
Ecommerce, fintech, and food delivery
Deliverable
production ML models
Required skills
Large-scale LLM training (pre-training, fine-tuning), distributed training frameworks (DeepSpeed, FSDP, Megatron-LM), data preparation at 100B+ tokens, Python, PyTorch, training stability debugging, evaluation framework design
Preferred skills
RLHF/DPO/GRPO, model quantization/distillation, GPU architecture optimization (A100/H100), synthetic data generation pipelines
Technologies
DeepSpeed, FSDP, Megatron-LM, Axolotl, PyTorch, Transformers, Accelerate, vLLM, TensorRT-LLM
Responsibilities
Design and execute experiments to improve model quality and efficiency; Drive technical decisions for model architecture and infrastructure; Mentor engineers and conduct code reviews; Train LLMs on proprietary datasets; Build and optimize distributed training infrastructure; Curate high-quality synthetic data; Debug training stability and optimize throughput; Build robust evaluation frameworks
Seniority
Senior, hands-on IC with technical leadership
