Principal AI/ML Architect: Discovery & Recommendations

Vumedi

Oakland (CA)

Hybrid

USD 180,000 - 250,000

Full time

14 days+
Application generator

A complete application in a minute — tailored resume and cover letter, ready to send.

Get past ATS filters

Job summary

Vumedi is seeking a Principal AI/ML Engineer to lead the architecture and development of its AI-powered discovery platform. You’ll shape how physicians access educational content through intelligent retrieval, semantic search, and personalized recommendations, working with our CTO to define technical strategy and production systems that scale globally.

This is a rare opportunity to apply modern AI techniques to improve patient care at scale and influence the platform’s AI roadmap across the

Qualifications

  • Production experience building LLM-powered applications.
  • Experience with retrieval pipelines, embeddings, and vector databases.
  • Knowledge of modern AI frameworks and orchestration tools.
  • Experience with distributed systems at meaningful scale.
  • Familiarity with cloud-native architectures.
  • Proficiency in Python and modern backend development.
  • Ability to prototype, evaluate, and iteratively improve models.

Responsibilities

  • Own the technical direction behind Vumedi’s next-generation discovery platform.
  • Design intelligent recommendation systems that personalize educational content.
  • Build LLM-powered discovery experiences that understand clinical intent.
  • Develop semantic search, retrieval, ranking, and pipelines using embeddings and vector search.
  • Design production-grade knowledge graph RAG systems.
  • Evaluate emerging AI technologies, frameworks, and infrastructure.
  • Build scalable inference and retrieval services for physicians worldwide.
  • Collaborate with Product, Data Science, and Engineering to iterate and launch new AI capabilities.
  • Define long-term AI architecture for the platform.

Skills

LLM-powered applications
Retrieval pipelines
Vector databases
Modern AI frameworks
Distributed systems
Cloud-native architectures
Python development
Experimentation & evaluation

Job description

Vumedi is seeking a Principal AI/ML Engineer to lead the architecture and development of its AI-powered discovery platform. You’ll shape how physicians access educational content through intelligent retrieval, semantic search, and personalized recommendations, working with our CTO to define technical strategy and production systems that scale globally.

This is a rare opportunity to apply modern AI techniques to improve patient care at scale and influence the platform’s AI roadmap across the

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Principal ML Engineer — AI-Driven Medical Discovery
Principal ML Engineer — AI-Driven Medical Discovery

VuMedi • Oak (NE)

Hybrid
USD 180,000 - 260,000
Principal ML Engineer, Recommendation Systems
Principal ML Engineer, Recommendation Systems

VuMedi • Oak (NE)

Hybrid
USD 180,000 - 260,000
Principal ML Engineer, Recommendation Systems
Principal ML Engineer, Recommendation Systems

Vumedi • Oakland (CA)

Hybrid
USD 180,000 - 250,000
Senior AI Solutions Architect – Platform & Delivery
Senior AI Solutions Architect – Platform & Delivery

Vizient • Chicago (IL)

On-site
USD 157,000 - 290,000
Principal Software Engineer - AI Discovery & Recommendations
Principal Software Engineer - AI Discovery & Recommendations

Harnham • Oakland (CA)

Hybrid
USD 250,000 - 350,000
Principal AI/ML Engineer for MedTech Platforms (Hybrid)
Principal AI/ML Engineer for MedTech Platforms (Hybrid)

Murata • Woodbury (MN)

Hybrid
USD 141,000 - 160,000
Medical, dental, vision
Paid time off
401k match
+4
Senior AI Solutions Architect & Platform Design Leader
Senior AI Solutions Architect & Platform Design Leader

Vizient • Edina (MN)

On-site
USD 157,000 - 290,000
Incentive eligible
Senior AI & Digital Discoverability Lead
Senior AI & Digital Discoverability Lead

Vertex Pharmaceuticals LLC • Boston (MA), Northern (KY)

Hybrid
USD 188,000 - 282,000
Annual bonus
Equity awards
Overtime pay eligibility
Senior AI & Data Science Architect
Senior AI & Data Science Architect

Vizient • Chicago (IL)

On-site
USD 118,000 - 206,000
Incentive Eligible
Benefits Plan
Principal AI Solution Architect, Platform & Delivery
Principal AI Solution Architect, Platform & Delivery

Vizient, Inc. • Irving (TX)

On-site
USD 157,000 - 290,000