Senior Or Staff ML Platform Engineer

Summary

Join TRM Labs, a blockchain intelligence company fighting crime globally. As an ML Platform Engineer, you'll design, develop, and deploy AI-driven solutions within a fast-paced, mission-driven environment. Collaborate with a multidisciplinary team of experts to build and enhance AI/ML model performance, ensuring data accuracy and reliability. You will build low-latency, high-throughput pipelines for real-time AI applications and integrate new AI technologies. This role requires strong coding skills (primarily Python), experience with various data engineering tools and technologies, and the ability to work with unstructured data. TRM Labs offers a remote-first work environment and opportunities for career growth.

Requirements

• Write high-quality code. We mostly work in Python
• Versatility. Experience across the entire spectrum of data engineering, including: Data stores (e.g., ClickHouse, ElasticSearch, Postgres , Redis, and Neo4j)
• Data pipeline and workflow orchestration tools (e.g., Airflow , DBT, Luigi, Azkaban, Storm)
• Data processing technologies and streaming workflows (e.g., Spark, Kafka, Flink)
• Deployment and monitoring infrastructure in public cloud platforms (e.g., Docker, Terraform, Kubernetes, Datadog)
• Loading, querying and transforming large data sets
• AI and ML Integration tools like MLflow, Feast, ONNX, Apache Beam
• You're comfortable working with noisy, dirty, and unstructured data to cleanse, scrape and convert it into structured data
• A high degree of initiative and ownership, combined with the ability to navigate ambiguity and adapt quickly to change
• Exceptional ability to structure problems and identify the most critical issues to prioritize
• Ability to communicate complex ideas effectively to both technical and non-technical audiences, verbally and in writing
• Experience working collaboratively in a cross-functional environment with a diverse group of people at all levels in an organization
• Passion for building a safer financial system for billions of people

Responsibilities

• Build and enhance AI/ML Model Performance
• Ensure data accuracy, consistency, and reliability, leading to better model training and inferencing
• Build low latency high throughput pipelines to enable real-time AI applications
• Enable AI-powered insights for quick decision-making
• Integrate and evaluate new AI technologies, accelerating innovation
• Help optimize operational costs while adding value
• Collaborate with data science, engineering, and business teams to shift focus on model building instead of data wrangling by embedding AI models into applications and platforms

Preferred Qualifications

Interest in virtual currencies, applications, and their use in financial networks and blockchain analysis is a plus

Benefits

• Purpose – Have a real-world impact, from disrupting terrorist networks to returning stolen funds
• Inspiring Colleagues – Collaborate with industry leaders and learn something new daily
• Career Growth – We embrace a growth mindset and provide development opportunities that increase skills relevant to your position
• Remote-first company