Staff AI/ML Engineer

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.

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.
• Proven track record implementing ML solutions such as classification systems, time series analysis, anomaly detection, or NLP applications that deliver measurable business impact.
• 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

Compensation:

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

#LI-NH1 #LI-REMOTE