Staff Software Engineer, Agent Engineering - US Remote

The Role

TRM Labs provides AI-powered intelligence solutions to help public and private sector agencies investigate and disrupt crime. Our AI Engineering Team builds robust pipelines, high-performance infrastructure, and operational tooling that allow AI systems to be deployed with speed, safety, and scale. We manage petabyte-scale pipelines, serve models with millisecond-level latency, and provide the observability and governance needed to make AI production-ready. We're also deeply involved in evaluating and integrating cutting-edge tools in the LLM and agent space.

What You Will Build

• 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 our LLM infrastructure (e.g. OpenAI, Anthropic, local models), including prompt engineering, RAG, and evaluation loops
• Design safe, observable, and auditable agent behaviors — ensuring 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

Requirements

• Strong engineering background with deep experience in backend or systems work (Python preferred)
• Hands-on experience building with LLMs, agents, and tooling frameworks (LangChain, semantic caches, vector DBs, etc.)
• 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 - you care about how agents impact real users (analysts) and optimize accordingly
• Bias toward experimentation and iteration - you’re excited to try, learn, and ship fast
• Previous experience with knowledge graphs, task orchestration, or AI safety a plus

About the Team

The AI Engineering team operates with a high level of collaboration and interdependence, fostering an inclusive culture where connections extend beyond work. We value diverse personalities and seek someone who is both results-driven and sociable, with strong communication, leadership, and teamwork skills, along with a positive, solutions-oriented mindset.

What We Offer

• Rapid Issue Resolution
• Navigating Bureaucracy
• Efficient Knowledge Transfer

Compensation

The estimated base salary range for this role is $200,000 - $275,000. Additionally, this role may be eligible to participate in TRM’s equity plan.

Company Culture

We are building a safer world. That promise shows up in how we work every day. TRM moves quickly. We are a high velocity, high ownership team that expects clarity, follow-through, and impact. People who thrive here are energized by hard problems, experimentation, and continuous feedback. If something takes months elsewhere, it will ship here in days. Our work sits at the intersection of AI, national security, and fighting crime. The problems are complex, the stakes are real, and the environment evolves quickly. The pace and intensity of the work reflect the importance of the mission. As a result, the way we operate requires a high level of ownership, adaptability, collaboration, and creative problem-solving.

Leadership Principles

• Impact-Oriented Trailblazer: We put customers first and move with speed, focus, and adaptability. We treat every plan like an experiment – test, ship, measure, and iterate quickly.
• Master Craftsperson: We care deeply about our craft. We balance speed with high standards, own outcomes end-to-end, and invest in getting better everyday.
• Inspiring Colleague: We add clarity and energy, not noise. We bring humility, candor, and a one-team mindset — giving and receiving feedback to make the team stronger.

Join Our Mission

At TRM, we care deeply about our craft. We are looking for individuals who want their work to matter, who experiment with speed and rigor, and who take pride in building a safer world for billions of people. If you’re excited by TRM’s mission but don’t check every box, we encourage you to apply — we hire for slope, judgment, and the will to learn fast.