Data Engineer - AI/ML

Role Description

TRM Labs provides AI-powered intelligence solutions that help public and private sector agencies investigate and disrupt crime. TRM's platforms enable investigators to trace illicit activity, build cases, and construct operating pictures of threat networks. Leading agencies and businesses worldwide rely on TRM to make the world safer and more secure.

This role keeps the government cloud environment fast, reliable, and compliant as it scales. You'll join the Data Platform Serving team, working on the StarRocks-backed serving layer that sits underneath government cloud investigations.

The impact you will have:

• Own performance tuning on the StarRocks serving layer, using AI-assisted query profiling to find and fix slow query patterns.
• Build and harden data pipelines feeding government cloud investigations, using AI code review workflows.
• Reduce single-point-of-failure risk on GovCloud data infrastructure by becoming the second engineer who can independently operate and troubleshoot the serving layer.
• Use AI-assisted debugging and log analysis to triage production issues in a regulated environment.

Qualifications

• U.S. citizenship is required for this role due to government cloud data access requirements.
• Hands-on experience operating distributed OLAP or serving-layer systems (StarRocks, Trino, ClickHouse, or similar).
• Experience owning data pipeline reliability and incident response.
• Comfort using AI tools (Claude, Cursor, or similar) to accelerate debugging, code review, and documentation.
• Independent ownership mindset with the ability to take on-call responsibility with minimal oversight.

Requirements

• Experience with query tuning and performance optimization at scale.
• Ability to pick up unfamiliar production infrastructure quickly.

Benefits

• High velocity, high ownership team.
• Opportunities for experimentation and continuous feedback.
• Work at the intersection of AI, national security, and fighting crime.

Company Description

TRM is a Series C company with $220M in total funding, backed by Blockchain Capital, 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.