Security ML Engineer — LLMs & Production Pipelines

Witness AI; United States; On-site; USD 36,000 - 60,000; Full time

Job summary

Witness AI is hiring a Machine Learning Engineer to design, build, and evaluate language models powering our security products. You will own end-to-end pipelines from data prep to experiment design, emphasizing reliable, interpretable, and safe models.

You will collaborate with researchers and data scientists to translate ideas into production systems, ensuring scalability and robust monitoring with CI/CD best practices.

Qualifications

• 2–5+ years in machine learning or data science.
• Experience with security-focused ML or infrastructure environments is a plus.
• Strong software engineering background with testing and CI/CD practices.

Responsibilities

• Build scalable pipelines for collecting and preprocessing data for LLMs.
• Design experiments to evaluate accuracy, robustness, and safety of models.
• Create dashboards and visualizations to communicate results.
• Develop knowledge graphs to enrich evaluation and context-based model performance.
• Collaborate with researchers and data scientists to integrate ideas into production workflows.

Skills

Python

PyTorch

CI/CD

Docker/Kubernetes

Data Engineering

Tools

Spark

Kafka

Airflow

Job description

Witness AI is hiring a Machine Learning Engineer to design, build, and evaluate language models powering our security products. You will own end-to-end pipelines from data prep to experiment design, emphasizing reliable, interpretable, and safe models.

You will collaborate with researchers and data scientists to translate ideas into production systems, ensuring scalability and robust monitoring with CI/CD best practices.