Head of Machine Learning Engineering
Role details
Job location
Tech stack
Job description
This is a senior leadership role sitting within one of Trainline's most important product areas: Core Experience. This pillar encompasses the products and systems closest to the customer journey - search, pricing, payments, fare intelligence, and conversational interfaces.
You will define and lead the ML strategy, team, and delivery across this domain. That means building and scaling ML and AI capabilities that measurably improve the customer experience and drive business outcomes. The work spans traditional ML systems and emerging agentic/LLM-based capabilities.
You will lead multiple ML teams through their respective team leads, reporting to the Director of AI & ML. You'll be accountable for the people (2 ML Managers and ~10 ICs), delivery, and budget across your pillar - including vendor and infrastructure cost management. You'll work closely with Product, Engineering, Data, Analytics, and commercial stakeholders, combining strategic thinking, technical leadership, and organisational leadership to ensure Trainline continues to ship high-quality, scalable ML systems in production.
As a Head of Machine Learning - Core Experience at Trainline you will...
- Set direction and own the budget. Define the vision, strategy, and roadmap for ML across the Core Experience pillar. Own the pillar's budget - including cloud infrastructure, vendor costs, and third-party tooling - balancing short-term delivery with longer-term platform investments.
- Lead and grow the organisation. Lead multiple teams through their team leads. Coach managers and senior ICs, set hiring strategy, and build a culture of technical excellence, experimentation, and delivery. Partner closely with Product, Engineering, and commercial stakeholders to identify high-value opportunities and raise adoption of ML across the business.
- Deliver and operate. Drive delivery across search relevance, payments models, splits and fare optimisation, supply automation, and conversational AI. Ensure your teams ship end-to-end - from prototyping through deployment, monitoring, and iteration. Raise operational maturity by defining SLIs/SLOs, incident management, and on-call practices for ML systems in production.
- Shape AI practices and governance. Co-own the operating model for modern AI at Trainline - LLM-based features, agentic systems, evaluation frameworks, and guardrails. Drive improvements in AI governance, data privacy, security, and audit accountability as capabilities scale., * Significant experience building and leading production ML teams, including managing through team leads (manager-of-managers), * Python and associated ML / DS libraries (scikit-learn, NumPy, LightGBM, Pandas, LangChain/LangGraph, TensorFlow, etc.)
- PySpark
- AWS cloud infrastructure: EMR, ECS, Athena, etc
- MLOps / DevOps: Terraform, Docker, Airflow, MLFlow
- Monitoring and evaluation platforms including New Relic and Braintrust
Requirements
Do you have experience in Terraform?, * Strong track record of shipping ML systems that deliver measurable business or customer impact, across problem types such as ranking, recommendation, forecasting, optimisation, or real-time decisioning
- Deep understanding of the full ML lifecycle and strong software/platform instincts - you know what it takes to run reliable ML at scale - including MLOps tooling and practices
- Experience raising operational standards: SLIs/SLOs, monitoring, incident management, and production reliability
- Budget ownership, including infrastructure cost management and vendor oversight
- Comfortable operating at both strategic and technical levels, with strong stakeholder and communication skills
- Experience hiring, mentoring, and developing technical talent at multiple levels
Nice to have
- Experience with AWS cloud infrastructure
- Experience with cross-functional delivery in product-led organisations
- Experience deploying conversational and agent-based systems in customer-facing products
- Experience in travel, transport, marketplaces, or similarly complex digital platforms
Benefits & conditions
Pulled from the full job description
- Private dental insurance
- Private medical insurance