AI Architect

About the role

The agent platform sits underneath every FourKites digital worker, and the teams building on it ship agents that track freight, chase suppliers, book dock appointments, answer customer questions and read shipping documents in production. You'll be the senior technical authority for how FourKites builds, ships and measures agents, working closely with agent teams in Chennai and with colleagues in the US and Europe.

What you'll do

• Design for long-running, multi-step work that spans many context windows and recovers cleanly when a tool, carrier API or model call fails
• Decide where each problem is best solved (frontier LLM, open-weight model, fine-tuned encoder, or plain code) and own model routing
• Build shared components for retrieval, compaction and summarization, sub-agent isolation, prompt caching and persistent memory so teams stop reinventing them
• Diagnose context failures (lost instructions, tool confusion, quality degradation as the window fills) and turn fixes into reusable harness patterns
• Build the offline eval framework: golden datasets from production traces, code-based checks, LLM-as-judge graders calibrated against human labels

What you need

• Machine learning
• LLMs
• Python
• AWS
• GCP
• Java
• Any graduate

Good to have

• TypeScript
• Kubernetes
• Kafka
• PostgreSQL
• Customer success
• Microservices