Staff ML Engineer

SupportFinity™; San Francisco (CA); Hybrid; USD 180,000 - 240,000; Full time

Benefits offered by this job

Equity

Unlimited PTO

Health, dental, and vision coverage

401(k)

Hybrid work model

Job summary

Knowtex is seeking a Staff ML Engineer to advance our voice AI and clinical NLP capabilities. You will work on cutting-edge problems in medical speech recognition and agentic AI systems for healthcare.

You will develop production-grade models, build NLP pipelines for clinical coding, and scale inference using Triton on AWS GovCloud, collaborating with clinicians to ensure alignment with MDM levels and quality measures.

Qualifications

• 5+ years of ML engineering with NLP/speech experience.
• Proficient in PyTorch or TensorFlow and transformer models.
• Skilled at building production ML pipelines at scale.

Responsibilities

• Develop models for medical speech recognition across 200+ specialties.
• Build clinical NLP pipelines for automated coding and ICD-10 classification.
• Implement note quality evaluation systems using LLMs and clinical rubrics.
• Scale inference infrastructure using Triton Inference Server on AWS GovCloud.
• Create specialty language models for gastroenterology, dermatology, and emerging markets.
• Design agentic AI systems for clinical decision support and documentation.
• Optimize model performance for real-time inference with sub-200ms latency.
• Collaborate with clinical teams to validate outputs against MDM levels.
• Build evaluation frameworks for MIPS quality measures compliance.

Skills

NLP

ML engineering

PyTorch/TensorFlow

Transformer models

Production ML pipelines

Cloud platforms

Model optimization

MLOps

Clinical NLP

ICD-10 familiarity

Education

Master's or PhD in CS/ML

Tools

AWS SageMaker

GCP Vertex AI

Triton Inference Server

Job description

About Knowtex

Knowtex is building the future of voice AI operating systems for clinicians, transforming how healthcare documentation happens at the point of care. Founded by Stanford AI scientists with deep clinical experience, we're experiencing explosive growth across both commercial health systems and federal healthcare, with our ambient documentation platform scaling rapidly to thousands of clinicians across hundreds of specialties. We're at an inflection point where cutting-edge AI meets real clinical impact, giving clinicians hours back each day to focus on what matters most - their patients.

Position Overview

We're seeking a Staff ML Engineer to advance our voice AI and clinical NLP capabilities. You'll work on cutting-edge problems in medical speech recognition, clinical language understanding, and agentic AI systems for healthcare.

Key Responsibilities

• Develop and optimize models for medical speech recognition across 200+ specialties
• Build clinical NLP pipelines for automated E&M coding and ICD-10 classification
• Implement note quality evaluation systems using LLMs and clinical rubrics
• Scale inference infrastructure using Triton Inference Server on AWS GovCloud
• Create specialty-specific language models for gastroenterology, dermatology, and emerging markets
• Design agentic AI systems for clinical decision support and documentation assistance
• Optimize model performance for real-time inference with sub-200ms latency requirements
• Collaborate with clinical teams to validate model outputs against MDM levels
• Build evaluation frameworks for MIPS quality measures compliance

Required Qualifications

• 5+ years experience in ML engineering with focus on NLP/speech recognition
• Strong expertise in PyTorch or TensorFlow
• Experience with transformer architectures and large language models
• Proficiency in building production ML pipelines at scale
• Understanding of model optimization techniques (quantization, distillation, pruning)
• Experience with cloud ML platforms (AWS SageMaker, GCP Vertex AI)
• Master's or PhD in Computer Science, ML, or related field

Preferred Qualifications

• Healthcare or clinical NLP experience
• Familiarity with medical terminology and clinical documentation
• Experience with speech recognition systems (Whisper, Conformer architectures)
• Knowledge of medical coding systems (CPT, ICD-10, SNOMED)
• Publications in ML/NLP conferences

Benefits

• Meaningful equity compensation
• Unlimited PTO
• Premium health, dental, and vision coverage
• 401(k) plan
• Hybrid work model: 3 days/week in our San Francisco office

About the company

Knowtex