← Back to jobs

Software Engineer (Python, AI)

Sundayy
FULL_TIME Remote · US US Posted: 2026-05-11 Until: 2026-07-10
Apply Now →
You will be redirected to the original job posting on BeBee.
Apply directly with the employer.
Job Description
About The Company TRM Labs is a leading provider of blockchain analytics and AI solutions dedicated to creating a safer world by combating crypto-related fraud and financial crime. Our mission is to empower law enforcement agencies, national security organizations, financial institutions, and cryptocurrency businesses with advanced tools to detect, investigate, and disrupt illicit activities within the digital asset ecosystem. Our platforms include comprehensive blockchain intelligence solutions capable of tracing fund sources and destinations, identifying suspicious activities, building investigative cases, and providing a holistic view of emerging threats. Trusted globally by top agencies and organizations, TRM Labs is committed to fostering a secure and transparent blockchain environment, supporting the integrity of financial systems and protecting users worldwide. About The Role We are seeking a highly skilled AI Engineer to join our innovative AI Engineering team. This role focuses on developing next-generation AI applications with an emphasis on Large Language Models (LLMs) and agentic systems. The successful candidate will be instrumental in architecting and implementing robust, scalable, and safe AI frameworks that facilitate tool use, context retrieval, memory management, and strategic planning within AI systems. You will be responsible for building intelligent, modular agents that automate investigative workflows and enhance analyst decision-making processes. Additionally, you will extend and optimize our LLM infrastructure, including prompt engineering, retrieval-augmented generation (RAG), and evaluation loops, ensuring high performance and reliability. Your work will directly impact TRM’s ability to innovate rapidly, deploy safe AI solutions, and maintain operational excellence in high-stakes environments. Qualifications Strong engineering background with extensive experience in backend or systems development, preferably in Python Hands-on experience with Large Language Models (LLMs), agent frameworks, and tooling such as LangChain, semantic caches, and vector databases Proficiency in designing and optimizing agentic pipelines and managing information flow into AI systems Deep understanding of system design principles emphasizing safety, scalability, and explainability High product empathy with a focus on user impact, especially for analysts and investigative teams Experience with experimentation, rapid iteration, and deployment of AI solutions Knowledge of knowledge graphs, task orchestration, or AI safety is a plus Comfortable working in a fast-paced, ambiguous environment with a collaborative mindset Excellent communication, leadership, and teamwork skills Responsibilities Architect and implement a robust, scalable agentic framework supporting tool integration, context retrieval, memory, and planning functionalities Build and maintain intelligent, modular agents that automate investigative tasks and assist analysts in decision-making Extend and optimize LLM infrastructure, including prompt engineering, retrieval-augmented generation, and evaluation processes Design and implement safe, observable, and auditable agent behaviors suitable for high-sensitivity environments Evaluate system performance using metrics such as reasoning accuracy, latency, success rate, and hallucination mitigation; iterate based on telemetry and user feedback Contribute to a culture of high ownership, rapid experimentation, and ethical AI deployment Collaborate cross-functionally with product, security, and research teams to ensure system robustness and compliance Stay abreast of cutting-edge advancements in LLMs, agent frameworks, and AI safety to inform continuous improvement Benefits Competitive salary package aligned with experience and skills Participation in TRM’s equity plan Comprehensive health, dental, and vision insurance Generous paid time off and leave policies Opportunities for professional development and continuous learning Collaborative and inclusive work environment in a rapidly growing company Flexible work arrangements and support for remote work where applicable Equal Opportunity TRM Labs is an equal opportunity employer committed to fostering a diverse and inclusive workplace. We do not discriminate on the basis of race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, disability, or any other protected status under applicable law. We believe that diverse perspectives and backgrounds drive innovation and excellence, and we welcome applicants from all backgrounds to join our miss