Browse Jobs
Explore open roles across India.Explore active software roles. Sign up to apply and get AI match scores.
Top MatchesTop Job Matches
Loading jobs…
Back to Browse Jobs
Abridge
Machine Learning Infrastructure Engineer- Model Inference
Abridge
San Francisco•On-site•2-5 Yrs•Jul 1
Job description
Overview
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.
Responsibilities
- 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.
Qualifications
- 2+ 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
- Excellent communication skills, with the ability to interface between research and product engineering
- 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
- Generous Time Off: 13 paid holidays, flexible PTO for salaried employees, and accrued time off for hourly employees.
- Comprehensive Health Plans: Medical, Dental, and Vision plans for all full-time employees. Abridge covers 100% of the premium for you and 75% for dependents. If you choose a HSA-eligible plan, Abridge also makes monthly contributions to your HSA.
- Paid Parental Leave: 16 weeks paid parental leave for all full-time employees.
- 401k and Matching: Contribution matching to help invest in your future.
- Pre-tax Benefits: Access to Flexible Spending Accounts (FSA) and Commuter Benefits.
- Learning and Development Budget: Yearly contributions for coaching, courses, workshops, conferences, and more.
- Sabbatical Leave: 30 days of paid Sabbatical Leave after 5 years of employment.
- Compensation and Equity: Competitive compensation and equity grants for full time employees.
- ... and much more!