Senior Machine Learning Engineer

This is a fully remote position, open to applicants in United Kingdom, +1 more state.

📋 Description

• Train, assess, and refine machine learning models and autonomous systems for customer feedback, including managing our bespoke fine-tuning pipelines. Conduct end-to-end experiments, meticulously track results, and provide clear recommendations on what to launch, enhance, or phase out.

• Develop and sustain LLM-driven features: retrieval pipelines, reranking systems, insight agents, data mining agents, and automated taxonomy generation.

• Create and implement robust evaluation frameworks: assemble test sets, define metrics, assess non-deterministic systems, manage class imbalance, and automate checkpoint comparisons.

• Enhance and expand semantic search and retrieval capabilities, transitioning from embedding-based strategies to more sophisticated methods.

• Write high-quality production code and collaborate closely with Engineering on productionization, model deployment, data pipelines, and monitoring processes.

• Partner with Product and Commercial teams to convert business requirements into viable ML solutions, and assist in client evaluations and accuracy assessments.

• Mentor team members, review code and research work, and integrate relevant advancements from the literature into our products.

⛳️ Requirements

• Profound understanding of transformer architectures.

• Strong proficiency in PyTorch, with the capability to create custom training loops, modify model architectures, and debug issues at the tensor level. Ideally, experience with parameter-efficient fine-tuning techniques such as LoRA.

• Extensive experience with large-scale, messy real-world text data, encompassing classification, extraction, embeddings, reranking, clustering, and search.

• Background in instruction fine-tuning and serving language models, with familiarity with frameworks like vLLM, DeepSpeed, or similar tools.

• A solid foundation in classical machine learning and statistics, with the ability to select simpler methods when appropriate.

• Practical experience in building with GenAI and agentic patterns.

• Strong communication skills and confidence in translating complex technical concepts for non-technical audiences (and vice versa!).

• Technical curiosity and a genuine interest in AI – a passion for experimenting to maximize the use of available technology.

• High sense of ownership and initiative, capable of identifying issues, prioritizing effectively, and driving solutions forward.

🏝️ Benefits

• Monthly Health & Wellness budget, increasing with length of service.

• Annual Learning and Development budget, increasing with length of service.

• Flexible working in a choice-first environment - we trust your preferred working style!

• Work From Home Allowance.

• 25 Holiday Days plus your local bank holidays, and an extra day for each year of service.

• Your birthday off.

• Enhanced Family Leave (UK Only), Fertility Leave, and Neonatal Leave.

• Optional Healthcare Plan.

• Life & income protection (Location dependent).

• Employee Assistance Programme (UK Only).

• The chance to partake in the company’s success through options.

• If you’re in London, a dog-friendly office with fantastic classes, events, and a rooftop terrace.