Staff AI Engineer - AI Infrastructure & Agentic Platform

The role

The Staff AI Engineer owns the AI infrastructure layer, including the organizational knowledge platform, agentic infrastructure on AWS, and the retrieval layer. You will architect and own the platform API surface and the cost-governance model.

What you will do

• Evolve the AI knowledge platform including retrieval, indexing, and synthesis layers.
• Architect and operate agentic infrastructure on AWS.
• Design and build graph-based, relationship-aware retrieval across data sources.
• Partner with product engineering to define the AI platform API surface.
• Build reference agent implementations for use cases like operational-incident triage and customer support.
• Own the AI infrastructure cost model by monitoring compute, model, and storage 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

• Inclusive and collaborative work environment
• Comprehensive health, vision, and dental coverage
• Benefits begin on your first day
• Generous PTO and company-paid holidays, including flexible floating holidays
• 100% 401(k) employer match up to 6%
• Paid parental leave
• Wellness support, including access to mental health resources

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.