Senior ML Infrastructure Engineer; Research Initiatives - Systems Integrator

Looking to architect next-generation AI infrastructure that transforms engineering simulations?

Join a world-class AI research laboratory, currently in Series B moving to C, that is redefining engineering with Large Physical Models that replace traditional numerical simulations with high-speed AI inference. The role involves building and managing the infrastructure required to train complex simulation models on hundreds of GPUs, leading the orchestration of massive-scale training environments, and contributing to pioneering Physical AI architectures that go far beyond standard generative models.

Engineers will work in a high-growth, NVIDIA-backed environment with exposure to cutting-edge AI hardware and sovereign cloud technologies.

Ready to design and scale infrastructure that powers the future of AI-driven engineering? Apply now.

• Design and scale training environments using PyTorch Distributed, JAX, or NVIDIA NeMo across multi-node/multi-GPU clusters.
• Manage a mixed-compute strategy spanning public clouds (AWS/Azure) and sovereign industrial clouds for sensitive data.
• Implement optimization techniques like FSDP and custom kernels to maximize FLOPS for irregular mesh and 3D geometric data.
• Build high-performance pipelines for ingesting CAE/CFD/FEA engineering data, ensuring zero I/O bottlenecks.
• Integrate traditional physics solvers (OpenFOAM/Simcenter) into ML pipelines for active learning and model refinement.
• Setup "physics-aware" CI/CD and experiment tracking (Kubeflow/MLFlow) that validates physical consistency laws.

• Orchestration:
  Expert-level Kubernetes (AKS/EKS) is essential.
• ML Frameworks:
  Strong proficiency in PyTorch, JAX, or NVIDIA NeMo.
• HPC/Data:
  Solid experience with Python, Go, and distributed data tools (Dask/Spark).
• Background:
  Experience in AI research labs (e.g., Deep Mind, OpenAI) or Neocloud environments.

• Competitive Equity:
  Significant stock option packages in a fast-scaling Series B/C firm.
• Flexible "London-Plus" Setup:
  Remote-first within Europe/UK, with roughly 1 week per month in London (all travel/accommodation fully paid).
• High-Impact Culture:
  Work alongside world-renowned physicists, mathematicians, and Formula 1 simulation veterans.
• Sponsorship:
  Full visa sponsorship available for top-tier global talent.

• Super Competitive (Base + Bonus + High-Upside Equity)
• Tailored to attract the best in the industry.