Remote Founding AI/ML Systems Architect

Bjak; Remote; GBP 80,000 - 100,000; Full time

Job summary

A leading AI group in the UK is seeking a founding technical builder to shape the core direction of their consumer AI application. This role involves designing training pipelines, model architectures, and scalable systems. The ideal candidate has extensive experience in deep learning, transformer models, and distributed training frameworks. With remote flexibility and competitive compensation, this position offers an opportunity to build impactful AI solutions in a collaborative environment.

Benefits

Competitive compensation

Performance-based bonuses

Remote-first flexibility

Insurance coverage

Qualifications

• Strong background in deep learning and transformer architectures.
• Hands-on experience training or fine-tuning large models.
• Proficiency with PyTorch, JAX, or TensorFlow.
• Experience with distributed training frameworks.
• Strong software engineering skills.

Responsibilities

• Build end-to-end training pipelines.
• Design new model architectures or adapt open-source models.
• Architect scalable inference systems.
• Shape early product direction and experiment with new use cases.

Skills

Deep learning

Transformer architectures

PyTorch

Distributed training frameworks

Software engineering

Tools

TensorFlow

JAX

GPU optimization

Job description

A leading AI group in the UK is seeking a founding technical builder to shape the core direction of their consumer AI application. This role involves designing training pipelines, model architectures, and scalable systems. The ideal candidate has extensive experience in deep learning, transformer models, and distributed training frameworks. With remote flexibility and competitive compensation, this position offers an opportunity to build impactful AI solutions in a collaborative environment.