Avatar for Collective Health
Collective Health
Actively Hiring
Applying technology and design to create the healthcare experience we all deserve
  • B2B
  • Scale Stage
    Rapidly increasing operations
  • Top Investors
    This company has received a significant amount of investment from top investors
  • +1

Senior Software Engineer in Test (AI Agentic Systems)

Posted: 2 months ago
Job Location
Lehi • 
Remote Work Policy

In office - WFH flexibility

Visa Sponsorship

Not Available

RelocationAllowed
Skills
Python
SQL
Pandas
Big Query
Pytest
Prompt Engineering
Function Calling
Langgraph
Cloud Logging
Vertex AI Experiments
Auto-SxS
Custom Mocking Frameworks for External APIs (Vertex AI/ADK)
RAG (Vertex AI Search)

About the job

At Collective Health, we’re transforming how employers and their people engage with their health benefits by seamlessly integrating cutting-edge technology, compassionate service, and world-class user experience design.

This is not a traditional QA role. You will be the quality owner for an LLM-based multi-agent pipeline that autonomously adjudicates health insurance claims for self-funded plan sponsors. You are building a Three-Tier Evaluation Framework to ensure our Gemini-powered agents reason correctly, call tools accurately, and produce DOL-ready outcomes.

You will work at the intersection of Vertex AI, healthcare compliance, and high-scale data engineering. Your work directly determines whether claims are paid correctly and whether the company can withstand a Department of Labor (DOL) or state DOI audit. The stakes are real, the domain is hard, and the problems are genuinely novel.

What you'll do:

  • Outcome Evaluation (The "What")
    • Golden Set Governance: Build and maintain a versioned library of "Grounding Data" results by working with senior claims examiners to define "Ground Truth."
    • Model-as-a-Judge Automation: Design automated "LLM-grading-LLM" workflows using custom rubrics to score factual grounding and policy compliance.
    • Semantic Assertion Framework: Develop testing libraries that move beyond string matching to validate semantic equivalence and numerical accuracy in agent outputs.
  • Trajectory Evaluation (The "How")
    • Function-Call Auditing: Use Vertex AI traces to programmatically verify that mandatory tools (via MCP) were invoked with correct arguments.
    • Orchestration Logic Validation: Assert that agents respect defined priorities across the four architectural layers: Data & Knowledge, Orchestration, Agentic Reasoning, and Tooling.
    • Reasoning Trace Auditing: Ensure every autonomous decision is traceable to a specific SOP sentence and a live API data point.
  • Continuous Automated Regression (The "Always")
    • CI/CD Integration: Every prompt or model update in Vertex AI Prompt Management must trigger an automated regression run.
    • Auto-SxS: Own the automated pairwise comparison process to detect logic drift between "New" and "Production" agent versions.
    • Mocking & Resilience: Build a Vertex AI/ADK mocking layer to simulate model responses, allowing for thousands of logic tests in seconds with zero API costs.

To be successful in this role, you'll need:

  • Required Skills (The Core Bar)
    • Python SDET Expertise: Expert in Python and pytest, specifically building custom mocking frameworks for external APIs (Vertex AI/ADK).
    • AI/LLM Observability: Hands-on experience with Vertex AI Experiments, Auto-SxS, and Cloud Logging for trace analysis.
    • Data Literacy: Expert-level SQL (BigQuery) and Pandas skills to "diff" massive datasets and identify adjudication discrepancies.
    • Prompt Engineering for QA: Ability to analyze "System Instructions" and refine prompts based on failed test cases to close logic gaps.
    • Architectural Testing: Experience testing multi-layer systems involving RAG (Vertex AI Search), state management (LangGraph), and function calling.
  • Preferred Skills (The "Nice-to-Haves")
    • Healthcare/Claims Domain: Familiarity with claims adjudication concepts (pend reason codes, COB, eligibility, stop-loss).
    • Compliance Knowledge: Understanding of HIPAA/PHI handling and writing test evidence for regulatory bodies (DOL/DOI).
    • Human-in-the-Loop Testing: Experience in "Shadow Mode" monitoring—comparing agent decisions against human expert (MCA) baselines.

Pay Transparency Statement

This is a hybrid position based out of our Lehi office, with the expectation of being in office at least two weekdays per week. #LI-hybrid

The actual pay rate offered within the range will depend on factors including geographic location, qualifications, experience, and internal equity. In addition to the salary, you will be eligible for 115000 stock options and benefits like health insurance, 401k, and paid time off. Learn more about our benefits at https://jobs.collectivehealth.com/benefits/.

Lehi, UT Pay Range

$99,200—$124,000 USD

Plano, TX Pay Range

$109,120—$136,400 USD

Why Join Us?

  • Mission-driven culture that values innovation, collaboration, and a commitment to excellence in healthcare
  • Impactful projects that shape the future of our organization
  • Opportunities for professional development through internal mobility opportunities, mentorship programs, and courses tailored to your interests
  • Flexible work arrangements and a supportive work-life balance

We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status. Collective Health is committed to providing support to candidates who require reasonable accommodation during the interview process. If you need assistance, please contact [email protected].

Privacy Notice

For more information about why we need your data and how we use it, please see our privacy policy: https://collectivehealth.com/privacy-policy/.

About the company

Collective Health company logo

Collective Health

Actively Hiring
Applying technology and design to create the healthcare experience we all deserve501-1000 Employees
Company Size
501-1000
Company Type
Healthcare Technology
Company Industries
Health and Insurance
  • B2B
  • Scale Stage
    Rapidly increasing operations
  • Top Investors
    This company has received a significant amount of investment from top investors
  • Valuation $1B+
    This company has a valuation of $1B or more
Learn more about Collective Health image

Funding

AMOUNT RAISED
$437M
FUNDED OVER
5 rounds
Rounds
F
$280000000
Series F - May 2021+4

Perks

Financial & Commuter
We understand peace of mind is an essential benefit of financial fitness. That’s why we offer commuter assistance, flexible spending accounts, a retirement plan, and life insurance with disability coverage.
Retirement
401k
Time Off Program
We work hard but we also all need time out of the office for our personal goals. To that end, our time off programs are designed to support your needs, from vacation time to leaves of absence.
Workspace
We always want to promote your health and wellness at work. We provide ergonomic support, carefully selected snacks, break-out spaces, and more.
Health & Wellness
We know the importance of staying active. Our monthly wellness stipend can go towards a gym membership, personal training, nutrition counseling and wellness apps.
Commuter
Team Activities
If there is a community or fun activity that you can think of, there is probably a club for it at Collective Health. And if it doesn’t exist, you can start it!

Founders

Ali Diab
Founder
San Mateo
image
Rajaie Batniji
Founder
image
View the team image