Jobs · Engineering · New York

Senior / Staff Software AI Test Engineer, AI Engineering

TWG AI · New York, NY · 1 mo ago
On-siteEngineeringFull-time

Key Responsibilities

  • Framework and harness engineering
  • Design and build scalable, reusable test automation frameworks for AI agents, LLM-powered applications, and underlying APIs.
  • Write clean, maintainable Python for test harnesses, eval pipelines, synthetic data generation utilities, and internal tooling.
  • Treat test code as production code: code review, type hints, documentation, library design.
  • Evaluation infrastructure
  • Build evaluation infrastructure for benchmarking agent performance against SOTA LLMs, competitors, and internal baselines.
  • Own regression suites, golden datasets, rubric-based evals, and metric dashboards.
  • Build tooling for synthetic test data generation, edge-case discovery, and adversarial testing.
  • Resilience and load
  • Design and run release, system, performance, and load tests against streaming, stateful, and async systems.
  • Build chaos and fault injection tooling for token expiry, connection pool exhaustion, provider failover, and cache pressure scenarios.
  • Drive contract testing across LLM providers (Bedrock, Anthropic, OpenAI) to catch parity drift.
  • CI/CD and observability
  • Integrate automated tests into CI/CD so every model, prompt, and code change is validated before it ships.
  • Build trace-based assertions on LangGraph state, tool calls, and agent decisions — debugging an agent failure means replaying graph state, not re-running a prompt.
  • Make observability a first-class testing surface (LangSmith, audit logs).
  • Human-in-the-loop and partnership
  • Implement HIL review workflows where automation alone cannot validate quality, then push the automation boundary outward.
  • Partner with AI engineers and data scientists on model evaluation, training and eval data prep, and root-cause debugging of complex end-to-end failures.
  • Champion quality engineering practices across the team: code review, coverage standards, observability, reproducibility.
  • Ensure user-centric validation so AI outputs are accurate, reliable, and meet real-world application needs.

Requirements

  • 3-7 years of software engineering experience, with a meaningful portion focused on test automation, SDET, or software engineering in test roles.
  • Expert-level Python. You write Python every day, design libraries other engineers use, and apply OOP and clean-code practices.
  • Hands-on Java experience, enough to read, write, and test Java services, not just touch them.
  • Working understanding of the LangGraph or Vercel frameworks: graph state, nodes, edges, tool calls, and how to write evals against agentic flows.
  • Demonstrated experience building eval sets for LLM models (this is critical to the role).
  • Experience testing across multiple client surfaces: iOS apps, plugins, and Chrome extensions.
  • Hands-on experience building automated test suites with frameworks such as pytest, Selenium, Playwright, Cypress, or similar.
  • Proven experience integrating test automation into CI/CD systems (GitHub Actions, Jenkins, CircleCI, GitLab CI, or similar).
  • Strong skills in data manipulation, test data preparation, and SQL.
  • Bachelor's degree or higher in Computer Science, Engineering, or a related field.

Similar jobs