机器学习基础设施工程师，模型推理
Machine Learning Infrastructure Engineer, Model Inference
关于ABRIDGE
ABRIDGE成立于2018年，使命是推动医疗领域的深入理解。我们的AI驱动平台专为医疗对话设计，提升临床文档效率的同时，让临床医生专注于最重要的事情——他们的患者。
我们的企业级技术可实时将患者与临床医生的对话转化为结构化临床笔记，并深度集成电子病历系统。依托于Linked Evidence和我们专为审计设计的AI，我们是唯一一家将AI生成的摘要映射到真实数据的公司，帮助医护人员快速信任并验证输出结果。作为医疗领域生成式AI的先驱，我们正在为AI在医疗系统中的负责任部署设定行业标准。
我们是一支不断壮大的团队，包括执业医生、AI科学家、博士、创意人员、技术人员和工程师，共同致力于赋能人们，让医疗更有意义。我们设有旧金山Mission区、纽约苏荷区和匹兹堡East Liberty的办公室。
职位描述
作为ABRIDGE的机器学习基础设施工程师（模型推理方向），你将在构建和优化支撑我们机器学习模型的核心推理基础设施方面发挥关键作用。你的工作将对提升我们AI驱动解决方案的可扩展性、效率和性能起到重要作用。你将与我们的基础设施和研究团队合作，构建、部署、优化并在我们的AI模型之间进行编排。
你将负责：
- 设计、部署和维护用于AI模型推理和训练的可扩展Kubernetes集群
- 开发、优化和维护机器学习模型服务基础设施，确保高性能和低延迟
- 与机器学习和产品团队合作，扩展AI驱动产品的后端基础设施，重点在于模型部署、吞吐量优化和计算效率
- 优化计算密集型工作流程，提高GPU在机器学习工作负载中的利用率
- 构建一个强大的模型API编排系统
- 与领导层合作，定义并实施随着公司增长而扩展基础设施的策略，确保长期的效率和性能
我们希望你具备：
- 在生产环境中构建和部署机器学习模型5年以上经验
- 对容器编排和分布式系统架构有深入理解
- 精通Kubernetes管理，包括自定义资源
查看英文原文
ABOUT ABRIDGE
Abridge was founded in 2018 with the mission of powering deeper understanding in healthcare. Our AI-powered platform was purpose-built for medical conversations, improving clinical documentation efficiencies while enabling clinicians to focus on what matters most—their patients.
Our enterprise-grade technology transforms patient-clinician conversations into structured clinical notes in real-time, with deep EMR integrations. Powered by Linked Evidence and our purpose-built, auditable AI, we are the only company that maps AI-generated summaries to ground truth, helping providers quickly trust and verify the output. As pioneers in generative AI for healthcare, we are setting the industry standards for the responsible deployment of AI across health systems.
We are a growing team of practicing MDs, AI scientists, PhDs, creatives, technologists, and engineers working together to empower people and make care make more sense. We have offices located in the Mission District in San Francisco, the SoHo neighborhood of New York, and East Liberty in Pittsburgh.
THE ROLE
As an ML Infrastructure Engineer, Model Inference 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.
STAYING SAFE - PROTECT YOURSELF FROM RECRUITMENT FRAUD
We are aware of individuals and entities fraudulently representing themselves as Abridge recruiters and/or hiring managers. Abridge will never ask for financial information or payment, or for personal information such as bank account number or social security number during the job application or interview process. Any emails from the Abridge recruiting team will come from an @abridge.com http://abridge.com email address. You can learn more about how to protect yourself from these types of fraud by referring to this article https://consumer.ftc.gov/consumer-alerts/2023/05/scammers-are-hijacking-job-ads-heres-how-spot-fakes. Please exercise caution and cease communications if something feels suspicious about your interactions.