Staff AI Engineer - AI Infrastructure & Agentic Platform

The role

The Staff AI Engineer owns the AI infrastructure layer, including the organizational knowledge platform and agentic infrastructure. You will architect retrieval layers, agentic systems on AWS, and cost-governance models.

What you will do

• Evolving the AI knowledge platform including retrieval, indexing, and synthesis layers
• Architecting and operating agentic infrastructure on AWS
• Designing and building graph-based, relationship-aware retrieval
• Partnering with product engineering to define the AI platform API surface
• Building reference agent implementations for triage and customer support
• Owning the AI infrastructure cost model and monitoring spend

Required qualifications

• 7+ years of professional software engineering experience
• At least 2 years building and operating production AI/LLM application systems
• Experience with hybrid search, re-ranking, query transformation, context-window management, and evaluation of retrieval quality in production
• Working experience with agentic frameworks and multi-step reasoning loops
• Production-grade software engineering fluency in Go, Python, or TypeScript
• Hands-on experience operating AI workloads on a managed cloud AI platform (AWS Bedrock or Azure AI Foundry)
• Hands-on Terraform experience
• Familiarity with production observability for AI systems
• Understanding of HIPAA-style data-handling requirements in a regulated SaaS environment

Nice to have

• Bedrock preferred
• Prior healthcare experience is a plus

Benefits

• High-growth, friendly, and engaging work environment
• Opportunities for career development
• Additional vacation days
• Modern office in Warsaw’s Powiśle district with Vistula River views, recreational facilities, and great nearby restaurants
• Private medical package
• Sports card
• Life and accident insurance

About the company

ShiftKey is a healthcare workforce marketplace that connects facilities with licensed and certified professionals. The company is building toward being an AI-native company to mitigate healthcare staffing shortages.