[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.ResponsibilitiesArchitect and implement a robust agentic framework that supports tool use, context retrieval, memory, and planningBuild intelligent, modular agents that automate investigative tasks and augment analyst decision-makingExtend and scale our LLM infrastructure (e.g. OpenAI, Anthropic, local models), including prompt engineering, RAG, and evaluation loopsDesign safe, observable, and auditable agent behaviors — ensuring reliability in high-sensitivity environmentsEvaluate performance across metrics like reasoning, latency, success rate, and hallucination, and iterate based on user feedback and system telemetryContribute to a culture of high ownership, rapid experimentation, and ethical AI deploymentSkillsStrong 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 systemsThoughtful approach to system design, with an eye for safety, scalability, and explainabilityHigh product empathy - you care about how agents impact real users (analysts) and optimize accordinglyBias toward experimentation and iteration - you're excited to try, learn, and ship fastPrevious experience with knowledge graphs, task orchestration, or AI safety a plusBenefitsMay be eligible to participate in TRM’s equity planCompany OverviewTRM Labs is a software company that offers blockchain, transaction monitoring, and analytics to help financial institutions and governments. It was founded in 2018, and is headquartered in San Francisco, California, USA, with a workforce of 201-500 employees. Its website is https://trmlabs.com.Company H1B SponsorshipTRM Labs has a track record of offering H1B sponsorships, with 2 in 2026, 1 in 2025, 4 in 2024, 3 in 2023, 3 in 2022, 1 in 2021. Please note that this does not guarantee sponsorship for this specific role.