AI Engineer – DTC Platform Engineering

Sits within the DTC Platform Engineering organisation and helps deliver the shared platforms, engineering services, and foundational infrastructure that enable product teams to build, test, deploy, and operate software securely and reliably at scale. Joining the AI-Native Engineering Enablement pillar, the position focuses on production-grade solutions to build, deploy, and scale LLM-powered applications and agentic workflows. It bridges core software engineering, modern machine learning, and tangible business impact by moving generative AI capabilities out of experimental prototypes and into resilient, high-performance production systems for product engineering teams.

• Construct multi-step AI agents and orchestration workflows using frameworks such as LangChain, LlamaIndex, or LangGraph, applying agentic design patterns.
• Develop production features using foundation models via cloud AI infrastructure.
• Deploy and configure enterprise AI and agentic gateways to handle model routing, semantic caching, load balancing, rate limiting, and cost/token tracking.
• Design and secure the universal integration layer connecting AI models to enterprise databases, internal codebases, and APIs using the Model Context Protocol.
• Build semantic search, document ingestion, and Retrieval-Augmented Generation (RAG) pipelines using vector databases, embedding models, and graph databases, and implement vectorless approaches where required.
• Design and maintain advanced cognitive memory systems, including episodic memory, semantic memory, and conversational state management across sessions using graph and in-memory databases.
• Deploy, version, and monitor AI services using containerisation and cloud infrastructure, configure serving architecture based on business needs, such as real-time APIs (REST/gRPC), batch prediction jobs, or edge deployment, and secure model endpoints, manage access controls (IAM)
• And ensure data privacy compliance.
• Administer central hubs such as MLflow or Weights & Biases to track model versions, lineage, code, and artifacts, and manage cloud infrastructure costs by auto-scaling compute resources and selecting appropriate hardware (CPUs vs GPUs) for production workloads.
• Build Automated Evaluation (Auto-Eval) platforms and CI/CD testing harnesses to stress-test prompts, RAG pipeline outputs, and multi-agent systems at scale.
• Partner closely with product managers, engineering managers, SREs, Infosec team, and frontend/backend software engineers to deliver end-to-end user experiences.

• Bachelor's or Master's degree in Computer Science, Software Engineering, Artificial Intelligence, or a related field.
• 4+ years of software development experience, with a dedicated focus on building, testing, and shipping agentic workflows, memory systems, and LLM features to production for 2+ years.
• Proven track record of deploying stateful AI systems, complex tool-calling agents, or enterprise API gateway integrations.
• Hands-on experience with multi-agent design patterns, state machines (LangGraph), and stateful memory management tools.
• Practical knowledge of configuring AI gateways for routing, fallbacks, and cost tracking.
• Experience building evaluation harnesses using frameworks such as DeepEval or Ragas.
• High proficiency in Python (and/or TypeScript for node-based agent runtimes).
• Experience with vector stores (Pinecone, pgvector, Milvus, Qdrant) and embedding models.
• Experience with RESTful/GraphQL APIs, Docker, Kubernetes, and enterprise cloud infrastructure (AWS, GCP, or Azure).
• Well-versed with agentic design patterns.

• Rust or Java understanding is good to have.

PUMA is a global sports brand creating footwear, apparel, and accessories that inspire athletes and everyday movers. The PUMA Group owns PUMA, Cobra Golf, and stichd, operates in 120+ countries, and has around 22,000 employees worldwide. PUMA looks for SPEED & SPIRIT in candidates, defined by values that inspire employees to BE DRIVEN in performance, BE VIBRANT in sporting legacy, BE TOGETHER in team spirit, and to let individual talent and experience shine.