Technical Lead, Machine Learning (AI)

Location: Hybrid in London, United Kingdom

Company

Builds a proactive smart assistant for everyday users, bringing AI to conversations, errands, organization, and workflows.

What you will do

• Own end-to-end ML system execution, including data pipelines, training workflows, evaluation systems, inference architecture, and deployment.
• Fine-tune and adapt models using LoRA, QLoRA, SFT, DPO, and distillation.
• Architect scalable inference systems while balancing latency, cost, reliability, and safety.
• Design training-data systems and evaluation pipelines covering performance, robustness, safety, and bias.
• Lead production deployment, including GPU optimization, memory efficiency, latency reduction, and scaling policies.
• Collaborate with application engineering to integrate ML systems into backend, mobile, and desktop products.

Requirements

• Experience building or shipping real ML systems used by people, not only demos.
• Strong Python skills and experience with PyTorch or JAX.
• Experience with GPU-based training and inference systems.
• Comfort working with large models and understanding their failure modes.
• Strong production-grade coding skills, with a focus on system correctness.
• Self-direction, pragmatic judgment, ownership, and clear communication in small teams.

Culture & Benefits

• Work in a small, high-talent-density, hands-on team.
• Make decisions collectively and move at a rapid pace.
• Balance high-quality delivery with rapid learning and iteration.
• Interviews are conducted virtually and/or onsite.
• Expect a prompt hiring decision after three, and no more than four, interviews.

Hiring process

• Applications are evaluated by technical team members.
• Complete three, and no more than four, interviews if there is a potential fit.
• Interviews take place via virtual meetings and/or onsite.