WitnessAI Logo

WitnessAI

Machine Learning Engineer

Posted One Month Ago
In-Office or Remote
Hiring Remotely in CA
Mid level
In-Office or Remote
Hiring Remotely in CA
Mid level
Designs, builds, and evaluates language models for AI security products. Responsibilities include dataset curation, preprocessing, LLM experimentation, safety and robustness evaluation, dashboards, knowledge graphs, model fine-tuning, production integration, monitoring, CI/CD, and ML tooling. Collaborates with researchers and data scientists to automate quality assurance and guardrails while delivering reliable, scalable, and interpretable models.
The summary above was generated by AI

Job Title: Machine Learning Engineer
Location: Cairo
Type: Full-time
Team: Machine Learning

About Us

Witness AI invented intent-based AI security. While legacy tools monitor what users say to AI, we understand what they're trying to accomplish - stopping jailbreaks, data exfiltration, and shadow AI before damage occurs. We provide visibility into how employees and systems use AI - capturing prompts, responses, and agent activity - so security teams can monitor risk, investigate incidents, and enforce guardrails in real time.

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
  • Experience: 2–5+ years working in machine learning or data science, ideally in a security or infrastructure-heavy environment.

  • Technical Skills:

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

    • Proficiency in ML frameworks such as PyTorch.

    • Experience with data engineering tools (e.g., Spark, Kafka, Airflow).

    • Familiarity with deploying models on cloud platforms (AWS, GCP, or Azure) and containerized environments (Docker, Kubernetes).

    • Strong knowledge of ML fundamentals (supervised/unsupervised learning, deep learning, NLP).

  • Security Awareness: Interest or background in cybersecurity, adversarial ML, anomaly detection, or related fields.

  • Startup Mindset: Comfortable working in fast-moving, ambiguous environments with a focus on shipping and iterating quickly.

Nice to Have
  • Research or industry experience in adversarial ML, model robustness, or explainable AI.

  • Experience building interactive dashboards for model monitoring and visualization.

  • Contributions to open-source ML, NLP, or security projects.

Salary Range

$36,000-$60,000 (The exact salary will be determined based on the selected candidate’s location, qualifications, experience, and relevant skills.)

Similar Jobs

3 Days Ago
Easy Apply
Remote
Canada
Easy Apply
Junior
Junior
Big Data • Fintech • Mobile • Payments • Financial Services
Develop and scale production machine learning models for credit decisioning, including default prediction and approval decisions. Build training, decisioning, monitoring systems, data pipelines, features, and algorithms. Conduct research, develop prototypes, and collaborate with engineering, credit, and product teams to deliver machine learning products from inception through business impact.
Top Skills: Deep LearningGeneralized Linear ModelsGradient BoostingKubeflowMachine LearningNumpyPandasProbabilistic CalibrationPythonRayScikit-LearnSparkSQLXgboost
11 Days Ago
Easy Apply
Remote or Hybrid
Canada
Easy Apply
Senior level
Senior level
Artificial Intelligence • Cloud • Information Technology • Security • Social Impact • Software
Lead the evaluation, training, fine-tuning, and production integration of machine learning and large language models. Build prompt engineering and model evaluation pipelines, optimize Elasticsearch-backed AI workflows, benchmark commercial and open-source models, and monitor production performance. Collaborate with product, engineering, and security teams to deliver safe, scalable AI capabilities while identifying improvements to data pipelines and engineering processes.
Top Skills: Amazon BedrockAmazon SagemakerAnthropicElasticsearchHugging FaceMeta LlmsMistralOpenaiPythonPyTorchScikit-LearnTensorFlow
17 Days Ago
Easy Apply
Remote or Hybrid
Canada
Easy Apply
Expert/Leader
Expert/Leader
eCommerce • Healthtech • Kids + Family • Retail • Social Media
Own the direction and implementation of Babylist’s personalization domain, including recommendation, search, feed models, custom embeddings, and shared representations. Take ambiguous problems from concept through production, owning architecture, deployment, monitoring, retraining, evaluation, and customer impact. Partner with product, design, and data, establish AI development standards, make cross-team technical decisions, and coach senior engineers while remaining hands-on with complex machine learning systems.
Top Skills: AirflowAmazon Bedrock AgentcoreAWSAws SagemakerDatadogDbtDeep LearningGithub ActionsHexKotlinKubernetesLangchainLaunchdarklyMatrix FactorizationMlflowMySQLPackwerkPandasPythonPyTorchRuby on RailsReactRetrieval And RankingScikit-LearnSidekiqSigmaSnowflakeSwiftTerraformTypescriptWarpstreamWeaviateXgboost

What you need to know about the Toronto Tech Scene

Although home to some of the biggest names in tech, including Google, Microsoft and Amazon, Toronto has established itself as one of the largest startup ecosystems in the world. And with over 2,000 startups — more than 30 percent of the country's total startups — Toronto continues to attract new businesses. Be it helping entrepreneurs manage their finances, simplifying business operations by automating payroll or assisting pharmaceutical companies in launching new drugs, the city's tech scene is just getting started.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account