Staff AI/ML Engineer

📋 Job Overview

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 Dragos

Dragos is an Equal Opportunity Employer and considers applicants for employment without regard to race, color, religion, sex, orientation, national origin, age, disability, genetics, or any other basis forbidden under federal, state, or local laws. All new hires must pass a background check as a condition of employment.

🎯 The Role

About the Role: 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). Troubleshoot and optimize ML model performance in production environments, addressing issues related to latency, accuracy, and resource utilization.

✅ Key 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).
• Troubleshoot and optimize ML model performance in production environments, addressing issues related to latency, accuracy, and resource utilization.

📌 Required Qualifications

• 6+ years of engineering experience with at least 4 years focused on machine learning implementations in production environments.
• Strong software engineering foundation with expertise in Python and SQL as well as experience with at least one additional language (Go, Rust, Java, or JVM-family languages).
• Demonstrated experience building and deploying ML systems using modern frameworks and libraries (scikit-learn, PyTorch, TensorFlow, HuggingFace, or similar).
• Experience with LLMs, retrieval-augmented generation (RAG), or advanced NLP techniques is beneficial.
• Experience with MLOps practices, including model versioning, monitoring, pipeline orchestration, and deployment in high-reliability environments.
• Familiarity with data engineering concepts, including data pipelines, stream processing, message queuing, and working with medium-to-large scale datasets.
• Knowledge of containerized deployment solutions and cloud-native architectures.
• Strong communication skills with the ability to explain technical concepts to diverse stakeholders and collaborate effectively across teams.
• Cybersecurity domain knowledge, particularly in threat detection, threat intelligence, or ICS/OT operations is preferred

🎁 Benefits

• Salary: $225,000
• Competitive Equity Package
• Comprehensive Benefits Plan

🛂 Visa & Eligibility

Dragos is an Equal Opportunity Employer and considers applicants for employment without regard to race, color, religion, sex, orientation, national origin, age, disability, genetics, or any other basis forbidden under federal, state, or local laws.