AI/ML Engineer

At Dragos, the mission is personal. The systems we protect deliver the water you drink, power your home, and keep the hospitals your community depends on running. Those critical infrastructure systems that power our civilization around the world are under attack every day by adversaries. When those systems fail, people are immediately at risk. We are the global leader in xOT cybersecurity, combining technology, threat intelligence, and expert services. The people here chose this work because they understand what is at stake. Here, you will find a remote-first mission-driven team across North America, Europe, the Middle East, and APAC built on authenticity, transparency, and trust. If safeguarding the systems that protect your family, friends, and community is the kind of work that matters to you, you are in the right place. 

About the Role:

We're seeking an experienced Staff Machine Learning Engineer to join our Engineering team. In this role, you'll drive the design and implementation of production machine learning systems within the Dragos platform. Working closely with Data Engineers and product teams, you'll build and deploy AI/ML capabilities that enhance threat detection, automate security analysis, and deliver actionable intelligence for Industrial Control System (ICS) and Extended Operational Technology (xOT) cybersecurity applications. 

Responsibilities:  

• Design and implement production-grade machine learning systems that expand Dragos product capabilities, with consideration for both cloud and resource-constrained on-premises environments. 
• Build and optimize ML model architectures for ICS/OT cybersecurity use cases, including threat detection, asset classification, behavioral analysis, anomaly detection, and natural language processing systems.
• Develop robust data pipelines and ML workflows that integrate with existing data infrastructure, supporting both real-time and batch processing requirements.
• Collaborate with OT detection experts to translate research concepts and prototypes into scalable, production-ready ML systems.
• Partner with Data Engineers to establish data contracts and implement observability frameworks for ML pipelines, including monitoring, versioning, and deployment best practices.
• Contribute to ML infrastructure improvements, including automated testing frameworks, CI/CD pipelines, and deployment strategies for containerized environments (Kubernetes, Docker).
• Evaluate and adapt state-of-the-art ML research and open-source models to domain-specific cybersecurity applications.
• Troubleshoot and optimize ML model performance in production environments, addressing issues related to latency, accuracy, and resource utilization.