AI/ML Engineer: Agentic Workflows

We're looking for a Senior AI/ML Engineer to design and build the foundational AI layers, multi-step agentic workflows, and developer tooling for the broader engineering organization. You'll focus on the internal runtime for AI: implementing prompt frameworks, enforcing deterministic structured outputs, and deploying production-grade Human-in-the-Loop state automations — turning raw context into reliable, zero-hallucination agent-based actions. Requirements Must-Have Requirements Overall Experience: 7+ years in Backend Software Engineering and AI/ML Application Engineering Agentic Architecture: 2+ years engineering multi-step LLM workflows, autonomous agent loops, and multi-agent orchestration frameworks in production Developer Platform Enablement: experience building internal SDKs, tools, or structured pipeline abstractions used by other product engineering teams to ship AI features rapidly Proficiency in Multi-Step Agentic Workflows & Tooling Proficiency in Advanced Amazon Bedrock & Prompt Engineering Proficiency in Structured Output Pipelines & Performance Ingestion Proficiency in State Management, Automation Consistency & HITL Production fluency in Python Proficiency in TypeScript/Node.js for Vercel AI SDK and AWS edge workers Preferred Experience Vercel AI SDK, LangGraph, or LlamaIndex for agentic orchestration Reliable tool-calling (function-calling) interfaces for LLM-driven internal system navigation Short-term and long-term agent memory architectures with ultra-low-latency in-memory stores (Amazon MemoryDB / Redis OSS / Valkey) Amazon Bedrock APIs (foundational model invocation and chaining for agent loops) Programmatic prompt optimization, few-shot prompting, dynamic context injection, and prompt caching Amazon Bedrock Guardrails (safety gating, prompt injection mitigation, PII redaction) Type-safe schema enforcement with Pydantic (Python) or Zod (TypeScript) Ingestion workers consuming Amazon MSK (Kafka), SQS, and log sources Optimistic Concurrency Control (OCC) and async write-delegation to prevent PostgreSQL deadlocks Deterministic multi-step pipelines with AWS Step Functions or LangGraph Durable HITL workflows (pause prior to DB writes or API transactions, persist state, resume on approval) Idempotent execution layers for safe agent loop retries without record duplication Working knowledge of C# (.NET Core) or Java for enterprise pipeline integrations Benefits Remote work. 13 floating holiday. 15 vacation days per year completed. Good working environment. Every qualified candidate who meets the requirements outlined in the job description will be considered in this hiring process without distinction. Furthermore, Keystone AI is an equal opportunity employer. We wholeheartedly embrace our responsibility to make employment decisions without regard to race, age, marital or social status, national origin, disability, sex, gender identity or expression, or any other characteristic or group of candidates or employees unrelated to their qualifications and suitability for the position. Our management is committed to upholding this policy with respect. Originally posted on Himalayas