About the role

Staff AI Platform Engineer developing AI inference and agentic systems for Paytm Labs. Leading the architecture and platform building for autonomous AI agents' operation in fintech.

Responsibilities

About the Role
We are a small team of AI builders in Paytm Labs.
As a Staff AI Platform Engineer, you will work across inference and agentic systems. You will
contribute to Paytm's AI inference platform (Pi), serving internal teams and enterprise customers
running our own coding and domain-specific models (voice, vision, risk, fintech workflows) as
well as third-party models. You will also architect and build the platform that enables
autonomous AI agents to operate safely and reliably in production - the runtime, orchestration,
and developer tooling for agents to reason, plan, use tools, and execute complex multi-step
workflows, automating both software development and business processes.
You will work at the intersection of LLMs, distributed systems, and production fintech
infrastructure, helping define how inference and agentic AI are built and deployed across
payments, risk, fraud, collections, support, and developer experience.

8+ years of software engineering experience, with 3+ years in AI systems or LLM applications
Strong understanding of LLM-based agent architectures (ReAct, RAG, tool use, multi-agent systems)
Experience building highly reliable distributed systems
Proficiency in Python and experience working with modern LLM APIs or open-source models
Experience with or strong interest in model serving (vLLM, TensorRT-LLM, Triton)
Understanding of distributed systems: task queues, event-driven architectures, state management
Experience with cloud platforms (AWS, GCP) and containerized deployments
Strong understanding of security risks in agentic systems (prompt injection, privilege escalation, data leakage)
Demonstrated experience leading complex technical initiatives
Strong written and verbal communication skills