Staff AI Engineer
Core
Architecting and owning ShiftKey's AI infrastructure layer, including the organizational knowledge platform, agentic infrastructure, and retrieval systems for a healthcare workforce marketplace.
Role type
Staff AI Engineer (Infrastructure & Agentic Systems)
Builds
AI-native platform serving internal engineering tools and customer-facing capabilities for healthcare staffing.
Domain
Healthcare workforce marketplace / Regulated SaaS (HIPAA)
Deliverable
production ML models | infrastructure
Required skills
Retrieval engineering (hybrid search, re-ranking, HyDE, query transformation), Agentic frameworks (multi-step reasoning, tool use, cost governance), Production software engineering (Go, Python, TypeScript), Cloud AI platforms (AWS Bedrock), Infrastructure as Code (Terraform), Production observability for AI systems.
Preferred skills
Healthcare industry experience, Graph-based retrieval design.
Technologies
AWS, AWS Bedrock, Terraform, Go, Python, TypeScript, Semantic RAG, HyDE.
Responsibilities
Evolve retrieval and synthesis layer to an organization-wide platform, Architect and operate agentic infrastructure on AWS, Design graph-based relationship-aware retrieval, Partner to define AI platform API surface, Build reference agent implementations, Own the AI infrastructure cost model.
Seniority
Staff, high autonomy with organization-wide influence