Machine Learning Engineer

Job Title : Machine Learning Engineer

Location : SF Bay Area

Type : Full-time

Team : Machine Learning

About Us

WitnessAI is a fast-growing SaaS startup on a mission to enable enterprises to adopt AI, safely. We're building a product that provides security and governance guardrails for public and private LLMs.

The Role

As a Machine Learning Engineer, you'll design, build, and evaluate language models that power our AI security products. You'll own the end-to-end pipeline - from dataset curation and preprocessing to experiment design, evaluation, and visualization of results. This role blends engineering and applied research, with an emphasis on producing reliable, interpretable, and safe language models.

What You'll Do

• Build scalable pipelines to collect, preprocess, and manage datasets for training and evaluation of LLMs.
• Design and run experiments to evaluate LLMs on accuracy, robustness, fairness, and safety.
• Create dashboards, reports, and visualizations to communicate evaluation results, trends, and failure cases.
• Develop and leverage knowledge graphs to structure data, enrich evaluation, and improve context-driven model performance.
• Work with researchers to translate new ideas into engineering workflows, and with data scientists to automate QA checks and guardrails.
• Fine-tune, optimize, and integrate models into production systems with a focus on reliability, scalability, and monitoring and CI / CD best practices.
• Contribute to ML tooling and experimentation frameworks to accelerate iteration.

What We're Looking For

Strong software engineering background (Python, testing frameworks like pytest / unittest, CI / CD tools).

Nice to Have

Benefits :

Salary Range

$125,000-$160,000 (Bay Area)