Principal / Distinguished / Fellow – AI Software Architect (Compute / Frameworks)

Principal / Distinguished / Fellow – AI Software Architect (Compute / Frameworks) – UK / Europe (Flexible / Remote) – £180,000 – £220,000+ base + 30% bonus

My client’s specialist software and architecture team is looking to add a Principal / Distinguished / Fellow – AI Software Architect to support the development of next-generation compute and graphics IP.

This role is focused on defining how AI workloads are deployed, optimised, and executed across GPU and accelerator-based platforms.

You will operate at the intersection of AI frameworks, runtime systems, and hardware architecture — ensuring that software and hardware are aligned to deliver optimal performance.

AI Software Architect Role & Responsibilities:

• Define and drive AI software and compute strategy
• Evaluate and optimise frameworks such as PyTorch and related ecosystems
• Work closely with architecture teams to map workloads onto hardware platforms
• Improve performance through runtime, compiler, and system-level optimisation
• Support prototyping and development of AI workloads
• Influence how AI capabilities are integrated into broader compute platforms
• Collaborate across software, architecture, and product teams globally

AI Software Architect Essential Skills:

• Strong experience with AI/ML frameworks (PyTorch, TensorFlow, etc.)
• Deep understanding of:
  • Inference and deployment
  • Performance optimisation
• Experience working with GPU or accelerator-based systems
• Exposure to compiler stacks (MLIR or similar) is beneficial
• Ability to operate across software and hardware domains
• Strong communication and cross-functional collaboration skills

What This Isn’t

• Not a pure data science or model training role
• Not research-only — this is applied, production-focused work
• Not isolated software — this is about hardware-aware optimisation

Why This Role

• Direct influence on AI capability within next-generation compute platforms
• Opportunity to shape how software and hardware interact
• Senior-level role with significant technical ownership
• High-impact position within a growing and strategically important area

Salary and Package:

• £180,000 – £220,000+ base salary
• 30% bonus
• Total compensation: £234,000 – £286,000+
• Flexible / remote working model
• Pension
• Private medical
• Life assurance
• Income protection
• Training and development

My client can offer a 3-stage process consisting of a 1^st stage Video Call , 2^nd Stage Video Call and a 3rd Stage on-site Interview (meet the team and site tour). This process can be completed within 2-3 weeks (based on availability)