Waystar

Waystar

Senior ML Engineer

Lehi, Utah, USA · Senior · full-time

Sponsorship not specifiedDetected 64 days ago
PythonVector DatabasesGCPCloud PlatformsCI/CDMachine LearningTensorFlowPyTorchNLPLLMsRAGAgentic AIMLOpsHIPAAResearchCollaborationProblem SolvingAdaptability

About the role

  • Language Model (LM) Development & Fine-tuning: Research, select, and experiment with appropriate open-source Language Models (Large & Small) (e.g., Phi-3, Mistral, Llama, Nemotron-H families) for specialized tasks.
  • Continuous Improvement & Research: Establish effective feedback loops from end-user interactions and system logs to identify areas for model improvement.
  • Curate and expand training datasets, ensuring data privacy (PHI/PII masking) and legal compliance.

Responsibilities

  • Data Pipeline & Knowledge Base Construction: Design, implement, and optimize robust pipelines for ingesting, parsing, and extracting structured information from complex documents (leveraging OCR, document layout analysis, Named Entity Recognition (NER), and Relationship Extraction (RE)).
  • Develop rich, nested JSON schemas for representing structured data and ensure scalable storage Generate and manage high-quality vector embeddings for efficient retrieval-augmented generation (RAG) within a Vector Database.
  • Explore and implement knowledge distillation techniques to transfer capabilities from larger models to smaller, more efficient LMs.
  • Agentic System Design & Implementation: Build and maintain the core agentic framework, including the orchestrator that intelligently routes queries and coordinates interactions between various specialized LM tools.
  • MLOps & Deployment: Deploy, manage, and monitor LMs and agentic components on Google Cloud Platform (Google Cloud Platform) using services like Vertex AI, GKE, Cloud Functions, and Cloud Run.
  • Implement robust MLOps practices for continuous integration, continuous delivery (CI/CD), model versioning, and performance monitoring (latency, throughput, accuracy).
  • Stay abreast of the latest research in LMs, agentic AI, NLP, and document understanding, applying relevant advancements to our system.
  • Active use of artificial intelligence (AI) tools and techniques to enhance performance, drive innovation, and improve decision-making across business functions.
  • Ability to leverage AI tools and platforms to streamline workflows, improve decision-making, and drive innovation.
  • Design, implement, and optimize robust pipelines for ingesting, parsing, and extracting structured information from complex documents (leveraging OCR, document layout analysis, Named Entity Recognition (NER), and Relationship Extraction (RE)).

Requirements

  • Direct experience with document processing technologies such as OCR, layout parsing, Document AI, or custom information extraction from unstructured text.
  • Experience with Vector Databases (e.g., pgvector, Pinecone, Weaviate, Qdrant) and RAG architectures.
  • Proven experience with Language Models (LMs), including model selection, fine-tuning, and deployment.

Benefits

  • The Waystar platform supports more than 500,000 providers, 1,000 health systems and hospitals, and 5,000 payers and health plans.
  • Develop and integrate "tools" (specialized LMs and external APIs) that perform atomic medical necessity tasks, ensuring strict behavioral alignment and structured outputs.
  • Bachelor's or Master's degree in Computer Science, Machine Learning, Artificial Intelligence, or a related quantitative field.
  • 3+ years of professional experience in Machine Learning Engineering, with a strong focus on NLP.
  • Curiosity and adaptability in exploring emerging AI technologies, with a mindset for continuous learning and experimentation.

Equal opportunity

  • We celebrate, value, and support diversity and inclusion.
Apply directly at Waystar →Create a free account for alerts like thisView Waystar immigration profile

This listing is sourced directly from Waystar's careers page and normalized into a canonical job model.