Staff Software Engineer, Platform (AI)

Location: London, UK

Company

Scale AI develops reliable AI systems, high-quality data, and full-stack technologies for enterprise, government, and other critical applications.

What you will do

• Own complex context and memory problems end to end, from system design through production.
• Architect infrastructure for agents to retrieve, store, and reason over long-running and cross-session context.
• Build and maintain retrieval and memory evaluation methodologies, including rubrics and benchmarks.
• Set technical standards for architecture, evaluation practices, and failure handling.
• Collaborate with software and machine learning engineers across orchestration, oversight, and systems optimization workstreams.
• Diagnose severe production issues involving incorrect retrieval, stale or leaked tenant context, and degraded relevance at scale.

Requirements

• 8+ years of engineering experience and a multi-year record of owning production systems end to end.
• Experience with embeddings, vector search, retrieval-augmented generation, fine-tuning, or agent memory architectures.
• Ability to work across software engineering and applied machine learning.
• Proficiency in Python and experience with containers, cloud platforms, and CI/CD infrastructure.
• Experience establishing technical standards and influencing decisions through design reviews, technical writing, or mentorship.

Nice to have

• Experience scaling products at hyper-growth startups.
• Experience with agent orchestration frameworks and multi-agent systems in production.
• Contributions to open-source agentic AI projects.

Culture & Benefits

• Software and machine learning engineers work on a shared roadmap.
• Work spans agent evaluation and oversight, orchestration, tool-use infrastructure, systems optimization, and applied research.
• Inclusive equal-opportunity workplace with reasonable accommodations for applicants with disabilities.