SIGN IN
Staff AI/ML Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Dragos, Inc. · 2 days ago

Staff AI/ML Engineer

Dragos, Inc. is a global leader in xOT cybersecurity, dedicated to protecting critical infrastructure systems. The Staff AI/ML Engineer will design and implement production machine learning systems to enhance threat detection and automate security analysis within the Dragos platform.
Cyber SecurityNetwork SecuritySecurity

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

Qualification

Machine LearningPythonSQLGoRustJavaJVM-family languagesScikit-learnPyTorchTensorFlowHuggingFaceLarge Language Models (LLMs)Retrieval-Augmented Generation (RAG)Natural Language Processing (NLP)Classification SystemsTime Series AnalysisAnomaly DetectionMLOpsModel VersioningModel MonitoringPipeline OrchestrationContainerized DeploymentKubernetesDockerCloud-Native ArchitecturesData PipelinesStream ProcessingMessage QueuingCybersecurityThreat Detection

Required

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)
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

Preferred

Experience with LLMs, retrieval-augmented generation (RAG), or advanced NLP techniques is beneficial

Benefits

Competitive Equity Package
Comprehensive Benefits Plan

Company

Dragos, Inc.

company-logo
Dragos provides the most effective OT cybersecurity technology for industrial and critical infrastructure to deliver on our global mission: to safeguard civilization.

Funding

Current Stage
Late Stage
Total Funding
$438.2M
Key Investors
WestCapBlackRock,Koch Disruptive TechnologiesKoch Disruptive Technologies,National Grid Partners (NGP)
2026-06-18Acquired
2023-09-18Series D· $74M
2021-10-28Series D· $200M

Leadership Team

leader-logo
Robert Lee
Chief Executive Officer and Founder
linkedin
leader-logo
Kurt Gaudette
Senior Vice President, Intelligence & Services
linkedin
Company data provided by crunchbase