Head of Machine Learning Engineering

Introducing Machine Learning and AI at Trainline 👋

Machine learning and AI are central to how Trainline helps millions of customers make smarter, more sustainable travel choices every day. Our ML models and AI systems power critical parts of the platform - from search and recommendations to pricing intelligence, personalisation, digital marketing, and AI-driven customer support.

Our ML teams own the full delivery lifecycle, from ideation through to production. We work closely with stakeholders across the business to expand the reach and impact of ML and AI throughout Trainline.

The Role

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.

We'd love to hear from you if you have...🔍

Significant experience building and leading production ML teams, including managing through team leads (manager-of-managers)

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

Our technology stack 💻

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

More information:

Enjoy fantastic perks like private healthcare & dental insurance, a generous work from abroad policy, 2-for-1 share purchase plans, an EV Scheme to further reduce carbon emissions, extra festive time off, and excellent family-friendly benefits.

We prioritise career growth with clear career paths, transparent pay bands, personal learning budgets, and regular learning days. Jump on board and supercharge your career from day one!

We're operating a hybrid model and ask that Trainliners work from the office a minimum of 60% of their time over a 12-week period. We also have a 28-day Work from Abroad policy.

Our values represent the things that matter most to us and what we live and breathe everyday, in everything we do:

💭 Think Big - We're building the future of rail

✔️ Own It - We focus on every customer, partner and journey

🤝  Travel Together - We're one team

♻️ Do Good - We make a positive impact

We know that having a diverse team makes us better and helps us succeed. And we mean all forms of diversity - gender, ethnicity, sexuality, disability, nationality and diversity of thought. That's why we're committed to creating inclusive places to work, where everyone belongs and differences are valued and celebrated.

Interested in finding out more about what it's like to work at Trainline? Why not check us out on LinkedIn , Instagram and Glassdoor !

Find Jobs in United Kingdom on Arbeitnow