Founding ML Research Engineer

confidential; Greater London; Hybrid; GBP 140,000 - 160,000; Full time

Benefits offered by this job

Equity

Job summary

Generative Group is seeking a Founding Machine Learning Research Engineer to take research into robust engineering and scalable model training in London. You will directly train neural networks, optimize training loops, and build data pipelines across multiple GPUs.

You will work with AI researchers and domain experts in a fast-moving, early-stage environment. The role is office-based in London with limited focused home-working days, and a strong emphasis on practical collaboration and ambitious

Qualifications

• Direct experience training neural networks at scale with PyTorch.
• Experience with multi-GPU or multi-node training and performance optimization.
• Strong Python and production-quality software engineering practices.
• Typically six or seven-plus years' relevant experience in a mid-career role.
• High agency, rapid learning ability and comfort working in an ambitious early-stage team.

Responsibilities

• Train and optimise neural networks and build internal workflows and data pipelines.
• Improve training performance across GPUs and maintain experiment workflows.
• Work closely with AI researchers and domain specialists in a fast-moving early-stage environment.
• Contribute to model architecture, attention mechanisms, and evaluation or reproducibility practices.

Skills

PyTorch

Multi-GPU training

Python

Production-grade code

ML research

Tools

Git

Linux

Job description

About the company

We are a low-profile, frontier AI research lab developing machine learning systems for extreme physics, including plasma physics and multispecies fluid dynamics. Our technology is designed to accelerate research and development across semiconductors, fusion energy, aerospace, defence and advanced engineering.

We build models from the ground up using high-fidelity simulation and experimental data, enabling faster and more reliable predictions across complex physical systems. Our mission is to create positive-impact AI that helps engineers and researchers explore more designs, parameters and operating conditions than traditional methods allow.

The company is seed-funded and has a small, highly technical team of AI researchers. We value scientific ambition, engineering excellence, rapid learning, direct communication, high agency and practical collaboration. This is an opportunity to help shape the foundations of a new category of AI alongside a high-calibre technical team.

What the job entails

As Founding Machine Learning Research Engineer, you will help take the team’s work from research into robust engineering and scalable model training. This is an engineer-first role with a strong machine learning research component-not a pure infrastructure or platform position.

You will directly train and optimise neural networks, build internal workflows and data pipelines, and improve training performance across multiple GPUs, currently up to approximately 20 GPUs in parallel. You will work across PyTorch model code, training loops, runtime behaviour, hardware utilisation and experiment workflows. Depending on your strengths, you may also contribute to model architecture, attention mechanisms, scientific machine learning and evaluation or reproducibility practices.

You will work closely with AI researchers and domain specialists in a fast-moving early-stage environment, making pragmatic technical decisions and turning promising ideas into reliable systems. The role is predominantly office-based in London, with limited focused home working possible.

Key requirements / What we look for

• Direct experience training neural networks at meaningful scale, with strong PyTorch expertise.




• Experience with multi-GPU or multi-node training and GPU or hardware performance optimisation.




• Strong Python and production-quality software engineering practices.




• [Suggestion: experience with vision modelling, world modelling or comparable research environments.]




• Typically six or seven-plus years' relevant experience in a mid-career role.




• High agency, rapid learning ability and comfort working in an ambitious early-stage team.




• A Master's degree, PhD or specific university background is not essential.

Compensation & Benefits

• Base salary: £140,000-£160,000.




• Equity: approximately 0.25%-0.5%, depending on experience and salary.

• London-based, predominantly office-based working, with one or two focused home-working days possible.