Live opening · Posted 27 days ago

Senior AI Engineer

Amazon · Bangalore
Instahyre 5-9 yrs
You are 27 days behind. JobBeeper subscribers saw this role while it was still new.

At a glance

The key details from the original listing.

Posted 27 days ago
CompanyAmazon
LocationBangalore
Experience5-9 yrs
SourceInstahyre
Listed27 days ago

Your early-applicant advantage

Live timing from JobBeeper.

Live data
0 min from Instahyre publishing this role to us finding it
16 min median time from a role going live to a subscriber being told
6 hours subscribers had this role before this page existed
16,247 roles found in the last 24 hours — the newest are not on this site yet
Start your free trial →

About the role

Description supplied by the original job listing.

The core responsibilities for the job include the following:
Inference Optimization:
Drive TTFT below 400ms for multi-step agent pipelines.
Streaming optimization: the first token to the user while sub-agents are still running.
KV cache strategy, prompt compression, dynamic context window management.
Multi-provider routing: model selection by latency, cost, and task type across OpenAI, Anthropic, Gemini, and open-weight models.
Agent Architecture:
Design and implement Plan-Execute-Synthesize pipelines that run sub-agents in parallel DAGs, not sequential chains.
Build reliable orchestration on top of Temporal: retries, timeouts, partial failure recovery, idempotency.
Structured output enforcement: JSON schema validation, retry loops on malformed LLM output, graceful degradation.
Tool call design: schema design that LLMs actually follow reliably across providers.
Evaluation and Harness:
Own the evaluation framework end-to-end: ground truth datasets, automated scoring pipelines, and regression detection on every PR.
LLM-as-judge pipelines for qualitative output assessment.
Latency regression testing - p50/p95/p99 tracked across every deployment.
Adversarial test case design: ambiguous queries, missing data, conflicting sources, and malformed tool responses.
Infrastructure:
Model serving and cold start optimization.
Async worker architecture for parallel sub-agent execution.
Observability: trace every token, every tool call, every synthesis step.
Requirements:
You've built something that runs in production at a meaningful scale and you understand why it's fast (or why it isn't).
Our stack: Go + Temporal for orchestration, a Plan-Execute-Synthesize agent architecture, and an evaluation harness we use to measure every regression. The problems are hard. The latency bar is aggressive. The accuracy requirements are unforgiving.
Stack familiarity (we care more about depth than match): Go, Python, Temporal, Kafka, PostgreSQL, and Docker.
Strong signal:
You've worked on inference pipelines where TTFT was the primary metric and you moved it meaningfully.
You've built multi-step agent systems and you know where they break, not from reading papers but from watching them fail in production.
You've written eval harnesses from scratch and you have opinions about what makes a ground truth dataset actually useful.
You've debugged LLM non-determinism in production and built systems resilient to it.
You've worked with streaming LLM responses and built infrastructure around partial output handling.
Weaker signal (but not disqualifying):
You've fine-tuned models but haven't shipped inference systems.
You've used LangChain/LlamaIndex but haven't built the layer underneath.
Strong ML research background without systems exposure.

Experience
5-9 yrs

Get JobBeeper Mobile App

Never miss a job opening! Get instant job alerts on your phone.

Subscribers see fresh openings within minutes. Download the JobBeeper App on Google Play to get real-time push notifications and apply before anyone else.

⚡ Instant Push Alerts 🎯 Tailored Filters 🚀 Direct Employer Links
GET IT ON Google Play

More openings worth a look

Recently tracked roles with full details and direct application links.

6 roles
Good roles move before most people even see them. Tell JobBeeper what you want and get fresh matches delivered in minutes.
Start your free trial →
⚡ Get fresh job alerts 📱 Get App