Shiftkey
Staff AI Engineer - AI Infrastructure & Agentic Platform
United States · Staff+
Sponsorship not specifiedDetected 6 days ago
TypeScriptPythonGoAWSAzureTerraformLLMsRAGAgentic AIA/B TestingComplianceCustomer SupportHIPAAResearch
About the role
- We are past the experimentation phase but nowhere near done - and that middle stage is exactly why this role exists.
- The next layer of the platform is in active development.
- You will not be maintaining someone else's finished platform, and you will not be starting from a blank page.
Responsibilities
- Designing and building graph-based, relationship-aware retrieval across the organization's data sources, enabling multi-hop queries and letting agents accumulate organizational knowledge over time.
- Building reference agent implementations on the platform - operational-incident triage, customer support, and future agentic use cases - grounding each agent's reasoning in institutional knowledge.
- A retrieval stack is already in production, serving real workflows today - this is not a from-scratch build.
- You will inherit serious retrieval foundations and own architecture of everything that comes next.
Requirements
- Familiarity with production observability for AI systems: metrics, structured logging for model spend and latency, and evaluation harnesses to detect regressions.
Nice to have
- Retrieval engineering beyond the basics.
- Our stack already includes re-ranking and HyDE
- Working experience with agentic frameworks and multi-step reasoning loops - tool use, iteration control, cost governance, and model routing trade-offs.
- Production-grade software engineering fluency (strict typing, testing, async/concurrency, modern toolchain) in Go, Python, or TypeScript, with the ability to ramp into another quickly.
- Hands-on experience operating AI workloads on a managed cloud AI platform (AWS Bedrock or Azure AI Foundry), including the identity/secrets model and model access governance.
- Bedrock preferred given our AWS stack.
- Hands-on Terraform experience - able to author and provision new infrastructure independently, not just modify existing modules.
- Understanding of HIPAA-style data-handling requirements in a regulated SaaS environment.
Benefits
- Prior healthcare experience is a plus, not a requirement.
- Comprehensive health, vision, and dental coverage.
- Benefits begin on your first day.
- Generous PTO and company-paid holidays, including flexible floating holidays.
- Paid parental leave.
- Wellness support, including access to mental health resources.
Company info
- About ShiftKey
- ShiftKey is a platform that is disrupting the way healthcare facilities find licensed and certified professionals to fill available shifts.
- Leveraging marketplace dynamics and deep industry knowledge, the company is playing a vital role in mitigating America's healthcare staffing shortages, enabling direct connections between facilities and healthcare professionals.
- By offering the opportunity to work as much or as little as they choose and putting the power back into the hands of healthcare workers, ShiftKey is bringing more licensed professionals back into the workforce, a solution that is solving a major crisis in healthcare.
- For more information, visit www.ShiftKey.com http://www.ShiftKey.com.
- ShiftKey is a healthcare workforce marketplace operating in a regulated (HIPAA) environment, and we are building toward being an AI-native company.
Apply directly at Shiftkey →Create a free account for alerts like thisView Shiftkey immigration profile
This listing is sourced directly from Shiftkey's careers page and normalized into a canonical job model.