Junior Machine Learning Engineer

Overview

Join Trainline as a Junior Machine Learning Engineer to help shape the future of train travel. You’ll work in a cross-functional AI/ML team and deliver scalable models that impact pricing, search, and personalized experiences. You’ll handle end-to-end ML delivery, from data exploration to deployment, collaborating with data scientists, engineers, and product managers. This role offers hands-on growth in a fast-paced, sustainability-driven environment with a strong emphasis on learning and experimentation.

Pay / Benefits

• private healthcare
• dental insurance
• work from abroad policy
• 2-for-1 share purchase plans
• EV Scheme
• festive time off

Responsibilities

• Design and deliver scalable ML models with measurable business impact
• Own the full ML delivery lifecycle from data exploration to deployment and maintenance
• Collaborate with data scientists, software and product teams in cross-functional squads
• Create tools and libraries to accelerate ML product delivery and improve workflows
• Contribute to AI/ML community and promote rigorous learning and experimentation
• Work with stakeholders to design data-driven products using Trainline datasets

Key requirements

• Advanced degree in Computer Science, Mathematics or similar quantitative field
• Proficiency with Python and data libraries (Pandas, NumPy, Scikit-learn)
• Experience deploying ML models in production
• Expertise in predictive modeling, classification, regression, optimization or recommendation systems
• Experience with Spark
• Knowledge of DevOps technologies such as Docker and Terraform and ML Ops platforms like MLFlow
• Experience with agile delivery, CI/CD, and data manipulation/feature engineering
• Familiarity with statistical methodologies
• Good communication skills

• communication
• collaboration
• problem-solving
• Python and ML/DS libraries (scikit-learn, NumPy, LightGBM, Pandas, TensorFlow)
• PySpark
• AWS cloud infrastructure (EMR, ECS, Athena)

…