Waystar · 1 week ago
Senior ML Engineer II
United States
Full-time
Remote
Mid, Senior Level
3+ years expWaystar is a company that helps providers simplify healthcare payments through innovative cloud-based technology. They are seeking a Senior ML Engineer to develop robust AI systems, focusing on the entire ML pipeline, including data extraction and fine-tuning language models for high-impact solutions.
FinanceHealth CarePaymentsSoftware
Responsibilities
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
Research, select, and experiment with appropriate open-source Language Models (Large & Small) (e.g., Phi-3, Mistral, Llama, Nemotron-H families) for specialized tasks
Design and execute efficient fine-tuning strategies (e.g., LoRA, QLoRA, full fine-tuning) on curated, domain-specific datasets to achieve precise performance for tasks like coverage determination, code lookups, and policy rule application
Explore and implement knowledge distillation techniques to transfer capabilities from larger models to smaller, more efficient LMs
Build and maintain the core agentic framework, including the orchestrator that intelligently routes queries and coordinates interactions between various specialized LM tools
Develop and integrate "tools" (specialized LMs and external APIs) that perform atomic medical necessity tasks, ensuring strict behavioral alignment and structured outputs
Deploy, manage, and monitor LMs and agentic components on Google Cloud Platform (GCP) 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)
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
Stay abreast of the latest research in LMs, agentic AI, NLP, and document understanding, applying relevant advancements to our system
Work closely with subject matter experts, product managers, and other engineers to translate complex requirements into technical solutions and evaluate system performance
Qualification
Required
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
Proven experience with Language Models (LMs), including model selection, fine-tuning, and deployment
Strong proficiency in Python and familiarity with ML frameworks (e.g., PyTorch, TensorFlow, Hugging Face Transformers)
Solid understanding and hands-on experience with core NLP techniques and architectures, especially Transformers
Experience with cloud platforms, particularly Google Cloud Platform (GCP), including services like Vertex AI, Cloud Storage, and compute services
Familiarity with MLOps principles and tools for model serving, monitoring, and pipeline automation
Excellent problem-solving skills, attention to detail, and ability to work independently and collaboratively
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
Curiosity and adaptability in exploring emerging AI technologies, with a mindset for continuous learning and experimentation
Preferred
Hands-on experience building or contributing to agentic AI systems or multi-agent frameworks
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
Exposure to the healthcare domain, particularly understanding medical terminology, CPT/ICD codes, or regulatory documents
Benefits
Competitive total rewards (base salary + bonus, if applicable)
Customizable benefits package (3 medical plans with Health Saving Account company match)
We offer generous paid time off for our non-exempt team members, starting with 3 weeks + 13 paid holidays, including 2 personal floating holidays. We also offer flexible time off for our exempt team members + 13 paid holidays
Paid parental leave (including maternity + paternity leave)
Education assistance opportunities and free LinkedIn Learning access
Free mental health and family planning programs, including adoption assistance and fertility support
401(K) program with company match
Pet insurance
Employee resource groups
Company
Waystar
Waystar is a technology platform that provides healthcare revenue cycle management solutions.
1001-5000 employees
H1B Sponsorship
Waystar has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (2)
2024 (2)
2023 (2)
2022 (2)
2021 (1)
Funding
Current Stage
Public CompanyTotal Funding
$2.12B2025-09-11Post Ipo Secondary· $709.2M
2025-05-13Post Ipo Secondary· $557M
2025-02-20Post Ipo Equity· $800M
Leadership Team
Steven M. Oreskovich
Chief Financial Officer
Ryan Ayres
VP Sales - Health Systems and Hospitals - Western Region
Recent News
2026-01-16
2026-01-13
2026-01-12
Company data provided by crunchbase