Senior ML Infra Architect for Large-Scale AI Simulations

Hamilton Barnes Associates Limited; United Kingdom; Hybrid; GBP 90,000 - 130,000; Full time

Benefits offered by this job

Significant stock option packages

Remote-first working setup

Fully paid travel and accommodation

Full visa sponsorship

Job summary

A leading AI research firm in the UK is seeking a skilled professional to architect AI infrastructure that will redefine engineering through high-speed AI inference. The role includes designing multi-node GPU training environments and collaborating with experts in a vibrant culture. The ideal candidate will be proficient in Kubernetes and experienced in machine learning frameworks. The firm offers competitive equity, a remote-first setup with some in-person presence in London, and full visa sponsorship for top talent.

Qualifications

• Experience designing multi-node/multi-GPU clusters.
• Knowledge of optimization techniques like FSDP and custom kernels.
• Ability to build high-performance pipelines for engineering data.

Responsibilities

• Design and scale training environments using multiple frameworks.
• Manage compute strategies across public and sovereign clouds.
• Implement optimization techniques to maximize performance.
• Integrate physics solvers into machine learning pipelines.
• Setup CI/CD and experiment tracking for validation.

Skills

Expert-level Kubernetes (AKS/EKS)

Strong proficiency in PyTorch

Strong proficiency in JAX

Solid experience with Python

Solid experience with Go

Experience with distributed data tools (Dask/Spark)

Background in AI research labs

Tools

NVIDIA NeMo

OpenFOAM

Simcenter

Job description

A leading AI research firm in the UK is seeking a skilled professional to architect AI infrastructure that will redefine engineering through high-speed AI inference. The role includes designing multi-node GPU training environments and collaborating with experts in a vibrant culture. The ideal candidate will be proficient in Kubernetes and experienced in machine learning frameworks. The firm offers competitive equity, a remote-first setup with some in-person presence in London, and full visa sponsorship for top talent.