Skip to content
SnowflakeSnowflakeMenlo Park, CA

Staff AI Engineer - Cortex Code Quality

Staff AI Engineer builds and owns quality systems for Cortex Code AI coding agents, including agent strategy, experimentation pipelines, failure analysis, and cross-team alignment. Requires 8+ years shipping AI/ML software, proficiency in Python/TypeScript/Go, and expertise in LLM evaluation harnesses.

236k – 339k/yr
On-site8+ YOEML Engineering

About the role

Responsibilities

  • Own major pillars of the quality stack: tuning agent behavior to engage on next generation agentic coding tasks.
  • Design and evolve pipelines and tooling that support large-scale experimentation, error mining, and iteration on prompts/tools/workflows with clear before/after signals.
  • Lead postmortems on quality regressions; cluster failure modes; translate findings into a prioritized roadmap for engineering and modeling partners.
  • Align product, infra, and applied AI on what “good” means for critical customer workflows; mentor engineers and uplevel eval craft across the team.
  • Ensure quality systems are dependable in practice—reproducible runs, stable datasets, versioning, and operational clarity when things drift.

Requirements

  • Bachelor’s degree in Computer Science, Engineering, Statistics, or a related field. Master’s or higher preferred but not a requirement.
  • 8+ years of experience shipping AI/ML-backed software in production, including Staff-level ownership of technical direction, cross-team delivery, and mentoring.
  • Strong track record building and operating eval harnesses, measurement, and/or experimentation loops for LLM/agent systems—not only one-off benchmarks.
  • Proficiency in programming languages such as Python, TypeScript, Go (strong in at least two).
  • Exceptional communication skills: crisp writeups, constructive debate, and ability to influence without authority across engineering and product.

Nice to Haves

  • Experience with data engineering pipelines (dbt, Airflow), data modeling, data analysis, retrieval systems, and semantic layers.
  • Deep experience with agentic coding tools (IDE agents, CLI agents) and intuition for model strengths, failure modes, and prompting limits.
  • Background in data engineering (dbt, Airflow), analytics, retrieval / RAG, or semantic layers.
  • Prior work on LLM observability, safety/guardrails, or quality systems used as release gates in production.

Skills

PythonTypeScriptGoLLMsAI/MLdbtAirflowRAGEval HarnessesExperimentation Pipelines

Similar roles

ML Engineering jobs
Crusoe

Senior Staff Software Engineer, AI Model LifeCycle

CrusoeSan Francisco, CA +1

Builds and manages platforms for AI model lifecycles, focusing on fine-tuning, training pipelines, and reinforcement learning for LLMs. Requires 8+ years in AI, advanced degree, and hands-on experience with generative AI techniques.

238k – 318k/yr
On-site8+ YOEML Engineering
Datadog

Staff GenAI Engineer - Application Performance Monitoring

DatadogNew York, NY

Technical leader on Datadog's APM team building and deploying GenAI/ML models for agentic investigations, automated troubleshooting, and incident triaging. Requires 10+ years experience leading large-scale GenAI initiatives end-to-end in product environments.

234k – 234k/yr
Hybrid10+ YOEML Engineering
Datadog

Staff Software Engineer - ML Observability

DatadogBoston, MA +1

Lead development of LLM observability features at Datadog, building tools for monitoring, tracing, and evaluating AI systems in production. Requires expertise in distributed systems, GenAI applications, and model internals.

234k – 300k/yr
HybridML Engineering
Traba

Staff Software Engineer

TrabaNew York, NY +1

Build and scale Traba's applied AI platform for industrial supply chain staffing and operations. Architect core backend systems integrating frontier models and agents for automation, matching, and vetting while owning performance and infrastructure.

240k – 300k/yr
Hybrid7+ YOEML Engineering
Traba

Staff Software Engineer

TrabaNew York, NY +1

Build and lead Traba's agentic AI platform as a founding member of the Agents team. Architect orchestration, evals, model strategy, and integrations for autonomous agents in industrial supply chain workflows. Requires 7+ years engineering experience including production LLM/agent systems, with customer immersion and 0-to-1 build experience.

240k – 300k/yr
Hybrid7+ YOEML Engineering