Staff AI Engineer – AI Infrastructure, Agentic Platform

ShiftKey

Staff AI Engineer working on AI infrastructure for ShiftKey's healthcare staffing marketplace. Responsibilities include evolving AI knowledge platform and architecting AWS agentic systems.

Posted 7/16/2026full-timeRemote • 🇵🇱 PolandLead💰 PLN 29,600 - PLN 32,000 per monthWebsite

Core Competencies

Role fit

Core Competencies

Use this summary to align your resume positioning with the role.

Demonstrates expertise in architecting and operating AI infrastructure on AWS, with a strong focus on retrieval engineering, multi-step reasoning, and cost governance. Proficient in building production-grade AI applications and managing observability in regulated environments.

Highest-signal resume keywords

AI Infrastructure ArchitectureRetrieval EngineeringProduction-Grade Software EngineeringAWS Bedrock ExperienceTerraform Proficiency

ATS Keywords

Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills

GoPythonTypeScriptHybrid SearchRe-RankingQuery TransformationContext-Window ManagementCost GovernanceObservability MetricsStructured Logging

Tools & Technologies

AWSTerraformAI WorkloadsManaged Cloud AI PlatformEvaluation Harnesses

Industry Keywords

HIPAA Data HandlingRegulated SaaS EnvironmentHealthcare Experience

Tech Stack

Tools & technologies

AWSAzureCloudGoPythonTerraformTypeScript

About the role

Key responsibilities & impact

Evolving the AI knowledge platform - taking the retrieval, indexing, and synthesis layer (currently semantic RAG + re-ranking + HyDE) to an organization-wide platform serving both internal engineering tools and customer-facing capabilities.
Architecting and operating agentic infrastructure on AWS - multi-step, tool-using AI systems that plan, retrieve, and act on complex queries and operational events, with cost guardrails and observability built in from day one.
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. This is on our roadmap, not in production - you will define the approach.
Partnering with product engineering to define the AI platform API surface, translating infrastructure primitives into developer-ready abstractions.
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.
Owning the AI infrastructure cost model: monitoring compute, model, and storage spend, flagging anomalies, and proposing guardrails to keep workloads within defined budgets.

Requirements

What you’ll need

7+ years of professional software engineering experience, with at least 2 years building and operating production AI/LLM application systems - not research, not prototyping, not demos.
Retrieval engineering beyond the basics. Our stack already includes re-ranking and HyDE; we need someone who has worked at or above that level: 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 - 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.
Familiarity with production observability for AI systems: metrics, structured logging for model spend and latency, and evaluation harnesses to detect regressions.
Understanding of HIPAA-style data-handling requirements in a regulated SaaS environment. Prior healthcare experience is a plus, not a requirement.

Benefits

Comp & perks

Additional vacation days for better work-life balance.
Thoughtfully designed private medical package to take care of what matters most.
Sports card to fuel your active lifestyle.
Life and accident insurance for peace of mind.
Modern office in Warsaw’s Powiśle district with Vistula River views, recreational facilities, and great nearby restaurants.
A high-growth, friendly, and engaging work environment with opportunities for career development.