Staff AI/ML Engineer

DragosUnited StatesAI/ML

Apply on company site

Opens the official Dragos careers page

Salary

Not listed

Job type

Full-time

Posted

21 Jul 2026

Status

Open · checked 7d ago

About this role

The Staff AI/ML Engineer role at Dragos in the United States involves designing and implementing production machine learning systems to advance industrial cybersecurity capabilities. This position is well-suited for an experienced engineer with a strong background in machine learning and a passion for protecting critical infrastructure.

Role Overview

Dragos seeks an experienced professional to drive the creation and deployment of production machine learning systems within the main platform. The chosen candidate collaborates closely with data engineers and product teams to deliver AI and machine learning features that improve threat detection, automate security analysis, and provide intelligence for industrial control system and extended operational technology security.

Key Responsibilities

Design and implement production-grade machine learning systems for cloud and resource-constrained on-premises environments.

Build and optimize model architectures for cybersecurity use cases like threat detection, asset classification, behavioral analysis, anomaly detection, and natural language processing.

Develop robust data pipelines and workflows supporting real-time and batch processing.

Collaborate with operational technology detection experts to turn research prototypes into scalable production systems.

Partner with data engineers to establish data contracts and set up observability frameworks for machine learning pipelines.

Contribute to infrastructure improvements including automated testing frameworks, continuous integration, and containerized deployments using Kubernetes and Docker.

Evaluate and adapt state-of-the-art research and open-source models for cybersecurity applications.

Optimize production model performance regarding latency, accuracy, and resource utilization.

Requirements

Six plus years of engineering experience, with at least four years specifically focused on machine learning implementations in production environments.

Strong software engineering foundation with expertise in Python and SQL.

Experience with at least one additional language such as Go, Rust, Java, or JVM-family languages.

Demonstrated experience building and deploying machine learning systems using frameworks like scikit-learn, PyTorch, TensorFlow, or HuggingFace.

Proven track record implementing solutions like classification systems, time series analysis, anomaly detection, or natural language processing.

Experience with MLOps practices including model versioning, monitoring, pipeline orchestration, and deployment in high-reliability environments.

Familiarity with data engineering concepts, data pipelines, stream processing, and message queuing.

Knowledge of containerized deployment solutions and cloud-native architectures.

Strong communication skills for explaining technical concepts to diverse stakeholders.

Nice to Have

Experience with large language models, retrieval-augmented generation, or advanced natural language processing techniques.

Cybersecurity domain knowledge, particularly regarding threat detection, threat intelligence, or industrial control system operations.

Skills and Tools

Python

PyTorch

TensorFlow

SQL

Kubernetes

Docker

Java

Rust

Natural Language Processing

Retrieval-Augmented Generation

MLOps

Salary and Benefits

Base salary of $225,000

Competitive equity package

Comprehensive benefits plan

About the Company

Dragos operates as a global leader in extended operational technology cybersecurity, combining technology, threat intelligence, and expert services to protect vital infrastructure systems. The organization maintains a mission-driven culture centered on authenticity, transparency, and trust across various regions.

How to Apply

Interested candidates should visit the official Dragos careers page to submit an application for this position.

Skills and tech stack

Python PyTorch TensorFlow SQL Kubernetes Docker Java Rust NLP RAG MLOps

Questions candidates ask

What is the salary for the Staff AI/ML Engineer role at Dragos?

The base salary is $225,000, accompanied by a competitive equity package and comprehensive benefits.

How many years of experience are required for this position?

Candidates must have six plus years of engineering experience, with a minimum of four years focused specifically on machine learning implementations in production.

What programming languages are required?

The role requires expertise in Python and SQL, along with experience in at least one additional language such as Go, Rust, Java, or JVM-family languages.

Is remote work available for this position?

Dragos maintains a remote-first team environment across multiple regions, and the job listing includes remote workplace indicators.

How do I apply for the Staff AI/ML Engineer job?

You can apply directly on the official Dragos careers page.

Ready to apply?

The application is on the official Dragos careers page.

Apply on company site