AI Agent Engineer - US Remote

The role

You will architect and implement a robust agentic framework and build intelligent, modular agents. You will also extend and scale LLM infrastructure including prompt engineering, RAG, and evaluation loops.

What you will do

• Architect and implement a robust agentic framework that supports tool use, context retrieval, memory, and planning
• Build intelligent, modular agents that automate investigative tasks and augment analyst decision-making
• Extend and scale LLM infrastructure including prompt engineering, RAG, and evaluation loops
• Design safe, observable, and auditable agent behaviors
• Evaluate performance across metrics like reasoning, latency, success rate, and hallucination
• Iterate based on user feedback and system telemetry
• Contribute to a culture of high ownership, rapid experimentation, and ethical AI deployment

Required qualifications

• Strong engineering background with deep experience in backend or systems work
• Hands-on experience building with LLMs, agents, and tooling frameworks
• Comfort working with agentic pipelines and optimizing information flow into AI systems
• Thoughtful approach to system design with an eye for safety, scalability, and explainability
• High product empathy

Nice to have

• Python preferred
• LangChain, semantic caches, vector DBs, etc.
• Previous experience with knowledge graphs, task orchestration, or AI safety

Benefits

• Equity plan

Before you apply

• You will be evaluated on applied AI fluency during the interview process
• The use of AI tools of any kind during TRM interviews is not permitted without prior approval

About the company

TRM Labs provides AI-powered intelligence solutions that help public and private sector agencies investigate and disrupt crime. The company is a Series C company with $220M in total funding.