[Remote] Staff Software Engineer, Agent Engineering - US Remote
Note: The job is a remote job and is open to candidates in USA. TRM Labs is a company that provides AI-powered intelligence solutions to help agencies investigate and disrupt crime. They are seeking a Staff Software Engineer to architect and implement frameworks for AI applications, focusing on building intelligent agents and scaling LLM infrastructure.
Responsibilities
- 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
- Contribute to a culture of high ownership, rapid experimentation, and ethical AI deployment
Skills
- 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
Benefits
- May be eligible to participate in TRM’s equity plan
Company Overview
Company H1B Sponsorship