Waystar

Waystar

Senior ML Engineer

Lehi, UT · Senior

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

Stay score

odds of building a lasting career here

37Risky
Cap-exempt (no lottery)0
Sponsors this role80
Entry-level history0
PERM / green-card track0
Lottery odds40
Fits your clock70

Thin sponsorship signal and lottery-bound. A low-probability bet with your clock running. Prioritize cap-exempt roles and proven entry-level sponsors first.

Lottery odds assume a STEM candidate.

Personalize to your clock →

Employer immigration record

from this employer's Department of Labor filings

Files H-1B transfers

5 transfer filings in the last year, covering 5 workers. Median labor-condition decision: 7 days. An employer that already files transfers is one that can take over an existing H-1B.

Sourced from Department of Labor LCA, PERM and prevailing-wage disclosure data. Employer matching is by name, so figures may be split across an employer's legal entities. Absence of a filing means none appears in our copy of the data, not that none exists.

Community outcomes

No reports yet — be the first to help the next applicant.

About the role

  • We are seeking a highly skilled and innovative Machine Learning Engineer with a passion for building robust, efficient, and domain-specific AI systems using Language Models (LMs) and agentic architectures.
  • As a core member of the team, you will be instrumental in developing the entire ML pipeline, from sophisticated data extraction techniques to fine-tuning specialized LMs and orchestrating their interactions within a multi-agent framework.
  • This is a unique opportunity to apply state-of-the-art Generative AI and NLP techniques to a real-world, high-impact problem, leveraging the latest research in agentic AI and LMs to deliver economical and powerful solutions.

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.
  • 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.
  • 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).

Nice to have

  • 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.

Skills

  • Job Description Summary

Compensation

  • Competitive total rewards (base salary + bonus, if applicable)

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
  • Develop and integrate "tools" (specialized LMs and external APIs) that perform atomic medical necessity tasks, ensuring strict behavioral alignment and structured outputs.

Company info

  • We are deeply committed to living out our organizational values: honesty; kindness; passion; curiosity; fanatical focus; best work, always; making it happen; and joyful, optimistic & fun.
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.