Staff AI Engineer - AI Infrastructure & Agentic Platform

Stack

Responsibilities

• ▹Evolve the AI knowledge platform: take the retrieval, indexing and synthesis layer (currently semantic RAG, re-ranking and HyDE) to an organization-wide platform serving internal engineering tools and customer-facing capabilities
• ▹Architect and operate 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
• ▹Design and build graph-based, relationship-aware retrieval across the organization's data sources, enabling multi-hop queries and letting agents accumulate organizational knowledge over time (on the roadmap, not in production; you define the approach)
• ▹Partner with product engineering to define the AI platform API surface, translating infrastructure primitives into developer-ready abstractions
• ▹Build reference agent implementations on the platform (operational-incident triage, customer support and future agentic use cases), grounding each agent's reasoning in institutional knowledge
• ▹Own the AI infrastructure cost model: monitor compute, model and storage spend, flag anomalies and propose guardrails to keep workloads within defined budgets

Requirements

• ▹7+ years of professional software engineering experience, with at least 2 years building and operating production AI/LLM application systems (not research, prototyping or demos)
• ▹Retrieval engineering beyond the basics: 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 the 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
• ▹Located in Poland

Nice to have

• ▹Prior healthcare experience

Soft skills

What we offer

• ▹Monthly gross remuneration of PLN 29,600 to PLN 32,000 under an employment contract (umowa o pracę), depending on experience and skill level
• ▹Remote work or work from the Warsaw office
• ▹A high-growth, friendly and engaging work environment with career development opportunities
• ▹Additional vacation days for better work-life balance
• ▹Modern office in Warsaw's Powiśle district with Vistula River views, recreational facilities and nearby restaurants
• ▹Private medical package
• ▹Sports card
• ▹Life and accident insurance

About the company

ShiftKey is a healthcare workforce marketplace that directly connects healthcare facilities with licensed professionals to fill available shifts. It operates in a regulated (HIPAA) environment.