Waystar
Senior ML Engineer II
Atlanta, GA · Senior
Sponsorship not specifiedDetected 30 days ago
PythonSQLNoSQLAWSGCPAzureCloud PlatformsCI/CDKafkaMachine LearningDeep LearningTensorFlowPyTorchscikit-learnSparkAirflowData EngineeringNLPLLMsRAGAgentic AIMLOpsStatisticsHIPAA
About the role
- Waystar is trusted by 1M+ providers, 1K+ hospitals and health systems, and is connected to over 5K commercial and Medicaid/Medicare payers.
- Waystar products have won multiple Best in KLAS® or Category Leader awards since 2010 and earned multiple #1 rankings from Black Book™ surveys since 2012.
- For more information, visit waystar.com or follow @Waystar on Twitter.
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).
- Stay abreast of the latest research in LMs, agentic AI, NLP, and document understanding, applying relevant advancements to our system.
Nice to have
- Work closely with subject matter experts, product managers, and other engineers to translate complex requirements into technical solutions and evaluate system performance.
- Ph.D. preferred.
- Expertise in programming languages such as Python (with extensive experience in ML libraries like TensorFlow, PyTorch, Scikit-learn).
- Strong experience with cloud platforms (AWS, Azure, GCP) and their ML services.
- Proficiency in building and managing data pipelines using tools like Spark, Kafka, SQL, and NoSQL databases.
- Demonstrated experience with MLOps principles and tools (e.g., MLflow, Kubeflow, Sagemaker, Airflow).
- Excellent problem-solving skills and the ability to work independently on complex issues.
Compensation
- Competitive total rewards (base salary + bonus, if applicable)
Benefits
- Through a smart platform and better experience, Waystar helps providers simplify healthcare payments and yield powerful results throughout the complete revenue cycle.
- 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.
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.