Staff Agent Engineer - US Remote

As a Staff Agent Engineer at TRM Labs, you will drive the scoping and execution of one or more objectives that impact multiple teams across the AI Engineering organization. You partner with stakeholders to set the team vision and technical roadmaps for AI-powered agent systems. TRM Labs provides AI-powered intelligence solutions that help public and private sector agencies investigate and disrupt crime, and the AI Engineering Team is chartered with enabling next-generation AI applications with a special focus on Large Language Models (LLMs) and agentic systems.

What you'll 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 the LLM infrastructure, including work with OpenAI, Anthropic, and local models, prompt engineering, RAG, and evaluation loops
• Design safe, observable, and auditable agent behaviors to ensure reliability in high-sensitivity environments
• Evaluate performance across metrics like reasoning, latency, success rate, and hallucination, and iterate based on user feedback and system telemetry
• Contribute to a culture of high ownership, rapid experimentation, and ethical AI deployment
• Drive multiple OKRs/projects that span teams, ensuring system health and quality through operational reviews, testing strategies, and monitoring rigor
• Serve as a role model and mentor to the entire engineering organization

What you bring

• Strong engineering background with deep experience in backend or systems work, with Python preferred
• Hands-on experience building with LLMs, agents, and tooling frameworks such as LangChain, semantic caches, and vector databases
• 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 and understanding of how agents impact real users, with ability to optimize accordingly
• Bias toward experimentation and iteration; excited to try, learn, and ship fast
• Track record of driving scoping and execution of complex projects across multiple teams
• Demonstrated mentorship and cross-team collaboration skills
• Alignment with TRM's Leadership Principles: Impact-Oriented Trailblazer, Master Craftsperson, and Inspiring Colleague

Nice to have

• Previous experience with knowledge graphs, task orchestration, or AI safety

What we offer

• Base salary range of $200,000 to $275,000
• Eligibility to participate in TRM's equity plan
• US remote work

About TRM Labs

TRM Labs is a Series C company with $220M in total funding, backed by Goldman Sachs, Bessemer, Y Combinator, Thoma Bravo, and others. Headquartered in San Francisco, TRM operates as a distributed-first company with hubs in Los Angeles, San Francisco, New York, Washington D.C., London, and Singapore. The company provides AI-powered intelligence solutions that help public and private sector agencies investigate and disrupt crime.