Member of Technical Staff, Machine Learning

Fdetimes; San Francisco (CA); Vor Ort; USD 180.000 - 250.000; Vollzeit

Benefits dieser Stelle

14 paid holidays

Health insurance

HSA contributions

Parental leave

401(k) matching

Personal device allowance

FSA & commuter benefits

Sabbatical leave

Zusammenfassung

Abridge is hiring an ML Infrastructure Engineer to build and optimize the core inference infrastructure powering our AI models. You’ll collaborate with Infrastructure and Research to deploy, scale, and orchestrate model workflows across GPUs and clusters.

You’ll help design scalable Kubernetes architectures, implement low-latency model serving, and drive efficiency across distributed systems. Join a fast-growing team shaping healthcare AI solutions.

Qualifikationen

• 5+ years of experience building and deploying ML models in production environments.
• Deep understanding of container orchestration and distributed systems architecture.
• Expertise in Kubernetes administration, including CRDs, operators, and cluster management.
• Experience developing APIs and managing distributed systems for batch and real-time workloads.
• Excellent communication skills interfacing between research and product engineering.

Aufgaben

• Design, deploy and maintain scalable Kubernetes clusters for AI model inference and training.
• Develop, optimize, and maintain ML model serving infrastructure for high performance and low latency.
• Collaborate with ML and product teams to scale backend infrastructure for AI-driven products, focusing on deployment, throughput, and compute efficiency.
• Optimize compute-heavy workflows and enhance GPU utilization for ML workloads.
• Build a robust model API orchestration system.
• Collaborate with leadership to scale infrastructure as the company grows for long-term efficiency.

Kenntnisse

ML in production

Container orchestration

Kubernetes

APIs & distributed systems

Communication

Tools

NVIDIA Triton

VLLM

TRT-LLM

PyTorch

TensorFlow

Terraform

Ansible

Jobbeschreibung

About Abridge

Founded in 2018, Abridge is the first AI-native clinician intelligence platform organized around the patient, built for clinicians, and designed to help health systems coordinate the clinical, financial, and evidence-based decisions for every moment of care. Powered by purpose-built AI for healthcare, Abridge connects patients, providers, payers, and life sciences companies. This year, the platform will support more than 100 million patient-clinician conversations across more than 300 of the largest and most complex health systems in the U.S. With deep workflow integration, the Abridge clinician intelligence platform has been validated to support encounters in 28+ languages across specialties and care settings.

Abridge is setting the industry standard for the responsible deployment of Al across health systems with continual and transparent Al evaluation. Clinical decision support references and cites gold-standard sources of evidence, and features like Linked Evidence map AI-generated documentation to source data, helping clinicians quickly trust and verify the output.

Abridge was awarded Best in KLAS for Ambient AI in 2025 and 2026, in addition to other accolades, including TIME Best Inventions of 2024 and 2025, CNBC Disruptor 2025, and Fast Company’s Most Innovative Companies 2025.

We are a growing team of practicing clinicians, AI scientists, PhDs, creatives, technologists, and engineers working together to transform healthcare. We have offices located in San Francisco, New York, and Pittsburgh.

The Role

As an ML Infrastructure Engineer at Abridge, you’ll play a pivotal role in building and optimizing the core inference infrastructure that powers our machine learning models. Your work will be instrumental in enhancing the scalability, efficiency, and performance of our AI-driven solutions. You will work with our Infrastructure and Research teams to build, deploy, optimize and orchestrate across our AI models.

What You'll Do

• Design, deploy and maintain scalable Kubernetes clusters for AI model inference and training
• Develop, optimize, and maintain ML model serving infrastructure, ensuring high-performance and low-latency.
• Collaborate with ML and product teams to scale backend infrastructure for AI-driven products, focusing on model deployment, throughput optimization, and compute efficiency.
• Optimize compute-heavy workflows and enhance GPU utilization for ML workloads.
• Build a robust model API orchestration system
• Collaborate with leadership to define and implement strategies for scaling infrastructure as the company grows, ensuring long-term efficiency and performance.

What You’ll Bring

• 5+ years of experience in building and deploying machine learning models in production environments.
• Deep understanding of container orchestration and distributed systems architecture
• Expertise in Kubernetes administration, including custom resource definitions, operators, and cluster management
• Experience developing APIs and managing distributed systems for both batch and real-time workloads
• Excellent communication skills, with the ability to interface between research and product engineering

Ideally, You Have

• Expertise with model serving frameworks such as NVIDIA Triton Server, VLLM, TRT-LLM and so on.
• Expertise with ML toolchains such as PyTorch, Tensorflow or distributed training and inference libraries.
• Familiarity with GPU cluster management and CUDA optimization
• Knowledge of infrastructure as code (Terraform, Ansible) and GitOps practices
• Experience with container registries, image optimization, and multi-stage builds for ML workloads
• Experience orchestrating across ASR models or LLM models for building various GenAI applications

Why Work at Abridge?

At Abridge, we’re transforming healthcare delivery experiences with generative AI, enabling clinicians and patients to connect in deeper, more meaningful ways. Our mission is clear: to power deeper understanding in healthcare. We’re driving real, lasting change, with millions of medical conversations processed each month.

Joining Abridge means stepping into a fast-paced, high-growth startup where your contributions truly make a difference. Our culture requires extreme ownership—every employee has the ability to (and is expected to) make an impact on our customers and our business.

Beyond individual impact, you will have the opportunity to work alongside a team of curious, high-achieving people in a supportive environment where success is shared, growth is constant, and feedback fuels progress. At Abridge, it’s not just what we do—it’s how we do it. Every decision is rooted in empathy, always prioritizing the needs of clinicians and patients.

We’re committed to supporting your growth, both professionally and personally. Whether it's flexible work hours, an inclusive culture, or ongoing learning opportunities, we are here to help you thrive and do the best work of your life.

If you are ready to make a meaningful impact alongside passionate people who care deeply about what they do, Abridge is the place for you.

How we take care of Abridgers

• Generous Time Off: 14 paid holidays, flexible PTO for salaried employees, and accrued time off for hourly employees
• Comprehensive Health Plans: Medical, Dental, and Vision coverage for all full-time employees and their families.
• Generous HSA Contribution: If you choose a High Deductible Health Plan, Abridge makes monthly contributions to your HSA.
• Paid Parental Leave: Generous paid parental leave for all full-time employees.
• Family Forming Benefits: Resources and financial support to help you build your family.
• 401(k) Matching: Contribution matching to help invest in your future.
• Personal Device Allowance: Tax free funds for personal device usage.
• Pre-tax Benefits: Access to Flexible Spending Accounts (FSA) and Commuter Benefits.
• Lifestyle Wallet: Monthly contributions for fitness, professional development, coworking, and more.
• Mental Health Support: Dedicated access to therapy and coaching to help you reach your goals.
• Sabbatical Leave: Paid Sabbatical Leave after 5 years of employment.
• Compensation and Equity: Competitive compensation and equity grants for full time employees.
• ... and much more!

Equal Opportunity Employer

Abridge is an equal opportunity employer and considers all qualified applicants equally without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, veteran status, or disability.

We're committed to providing reasonable accommodations throughout the interview process. Once you submit your application, we'll follow up with details on how to request an accommodation for interviewing, completing any assessments, or otherwise participating in the selection process.