Senior Machine Learning Engineer (PyTorch)

Location: London Area, United Kingdom (Hybrid)

Company

A highly technical organisation focused on the engineering infrastructure behind large-scale machine learning.

What you will do

• Evaluate new machine learning frameworks and hardware through hands-on experimentation.
• Improve the speed and efficiency of training and inference on large-scale GPU infrastructure.
• Investigate complex machine learning workload performance issues where solutions are not immediately apparent.
• Collaborate with engineering teams across the organisation to extend the scale and capabilities of their machine learning workloads.

Requirements

• Strong PyTorch experience and solid software engineering fundamentals.
• Experience working with machine learning training and inference workloads.
• Ability to analyse and optimise performance on GPU compute infrastructure.
• Interest in exploring new machine learning frameworks and hardware.
• Availability to work in a hybrid setup in London Area, United Kingdom.

Nice to have

• Research or big tech background.

Culture & Benefits

• Exploration and experimentation are a core part of the role.
• Access to compute and scale beyond what most companies operate.
• Opportunity to remain deeply technical.
• Permanent contract with hybrid working in London.