Principal Machine Learning Engineer, Recommendations

Vumedi

Oakland (CA)

On-site

USD 200,000 - 260,000

Full time

3 days ago
Be an early applicant

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Job summary

Vumedi, the world’s leading video education platform for physicians, seeks a Principal ML Engineer to lead the architecture and development of its AI-powered discovery platform. You will shape technical strategy with our CTO, build LLM-powered experiences, and drive production systems that deliver relevant content at scale for doctors worldwide.

You will design intelligent recommendation systems, semantic search, and knowledge graph RAG components, collaborating with Product, Data Science, and

Qualifications

  • Experience building intelligent discovery systems at scale.

Responsibilities

  • Own the technical direction of the next-generation discovery platform.
  • Design and build LLM-powered discovery experiences that understand clinical intent.
  • Develop retrieval, ranking, and recommendation pipelines using embeddings and vector search.
  • Build scalable inference and retrieval services for global physicians.

Skills

Recommendation systems
Semantic search
Retrieval-Augmented Generation
Vector search
Knowledge retrieval
Personalized content discovery
AI-powered user experiences
Python
Distributed systems
Cloud-native
LLM-powered applications

Tools

LLM frameworks
Vector databases
Embeddings

Job description

About Vumedi: Vumedi is the world's largest video education platform for physicians, trusted by more than 250,000 doctors worldwide to stay current on the latest clinical research, surgical techniques, and medical innovations. Every day, physicians come to Vumedi looking for the right knowledge at the right moment, whether they're preparing for a procedure, researching a rare condition, or learning from experts across thousands of specialties. We're building the next generation of that experience. Our vision is to transform how doctors discover medical knowledge by combining large language models, retrieval systems, semantic search, and personalized recommendations into an intelligent learning platform unlike anything else in healthcare. If you've built recommendation systems, search platforms, ranking algorithms, or AI-powered discovery experiences, we'd love to talk.

The Opportunity

We're looking for a Principal ML Engineer to lead the architecture and development of Vumedi's AI-powered discovery platform. This isn't about adding an AI feature to an existing product. You'll help define how hundreds of thousands of physicians discover knowledge through intelligent retrieval, semantic search, personalized recommendations, and LLM-powered experiences. Working directly with our CTO, you'll have significant influence over both technical strategy and implementation: from evaluating emerging AI technologies to designing production systems that deliver highly relevant educational content at scale. This is a rare opportunity to apply modern AI techniques to a mission that directly improves patient care around the world.

What You'll Build

You’ll own the technical direction behind Vumedi's next-generation discovery platform, including:

  • Designing intelligent recommendation systems that personalize educational content for physicians.
  • Building LLM-powered discovery experiences that understand clinical intent and not just keywords.
  • Developing semantic search, retrieval, ranking, and recommendation pipelines using embeddings, vector search, and modern retrieval architectures.
  • Designing production-grade knowledge graph RAG systems that combine medical knowledge with generative AI.
  • Evaluating and implementing emerging AI technologies, frameworks, and infrastructure as the ecosystem rapidly evolves.
  • Building scalable inference and retrieval services capable of serving physicians around the world.
  • Partnering closely with Product, Data Science, and Engineering to rapidly prototype, iterate, and launch new AI capabilities.
  • Defining the long-term technical architecture for AI across the Vumedi platform.
What We're Looking For

We're less interested in checking every technology box than finding someone who has built intelligent discovery systems at scale. Ideal candidates have experience building one or more of the following:

  • Recommendation systems
  • Search relevance and ranking
  • Semantic search
  • Retrieval-Augmented Generation (RAG)
  • Vector search and embeddings
  • Knowledge retrieval systems
  • Personalized content discovery
  • AI-powered user experiences

You may have worked on products involving:

  • Content recommendations
  • Marketplace ranking
  • Search infrastructure
  • Feed ranking
  • Personalized discovery
  • Knowledge graphs
  • Enterprise search
  • AI assistants
Technical Experience

We're excited by candidates who have experience with many of the following:

  • Production experience building LLM-powered applications
  • Retrieval pipelines, embeddings, and vector databases
  • Modern AI frameworks and orchestration tools
  • Distributed systems operating at meaningful scale
  • Cloud-native architectures
  • Python and modern backend development
  • Experimentation, evaluation, and iterative model improvement

You don't need experience with every technology we use today. We care far more about your ability to solve difficult discovery and AI problems than familiarity with a specific framework.

Why This Role Is Different

Many companies are adding AI features. We're rethinking how physicians discover knowledge. Medical education presents one of the most challenging retrieval problems in AI:

  • Massive amounts of highly specialized content
  • Complex user intent
  • Clinical context
  • Constantly evolving medical knowledge
  • High expectations for relevance and trust

You’ll help define how AI can responsibly connect physicians with the information they need to improve patient care.

Why Work at Vumedi
  • Work on problems that matter: Your work helps physicians learn new procedures, adopt new treatments, and ultimately improve outcomes for patients worldwide.
  • Build something from the ground up: You'll shape the future of AI discovery at Vumedi rather than optimizing a small piece of an already mature platform.
  • Work directly with technical leadership: Partner closely with our CTO and engineering leaders on long-term architecture and product strategy.
  • Own meaningful technical decisions: From retrieval architecture to recommendation systems to AI infrastructure, you'll influence how the platform evolves for years to come.
  • Join a collaborative engineering culture: Work alongside engineers, product leaders, and data experts across our Oakland headquarters and engineering office in Zagreb, Croatia.

Learn more about Vumedi

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Principal Machine Learning Engineer, Recommendations
Principal Machine Learning Engineer, Recommendations

Vumedi Inc. • Oakland (CA)

On-site
USD 180,000 - 240,000
Director of Engineering
Director of Engineering

Vumedi • Oakland (CA)

Hybrid
USD 190,000 - 270,000
Hybrid work model
Director of Engineering
Director of Engineering

Vumedi Inc. • Oakland (CA)

Hybrid
USD 200,000 - 320,000
Hybrid work schedule
Office in Oakland
Profitable growth and ownership
Lead ML Architect for AI-Driven Medical Discovery
Lead ML Architect for AI-Driven Medical Discovery

Vumedi • Oakland (CA)

On-site
USD 200,000 - 260,000
Principal ML Engineer: AI Discovery & Knowledge Retrieval
Principal ML Engineer: AI Discovery & Knowledge Retrieval

Vumedi Inc. • Oakland (CA)

On-site
USD 180,000 - 240,000
Senior Machine Learning Engineer
Senior Machine Learning Engineer

Videa Health, Inc. • Town of Boston (NY)

On-site
USD 110,000 - 150,000
Flexible PTO
Competitive pay and equity
Collaborative work culture
Senior Machine Learning Engineer
Senior Machine Learning Engineer

VideaHealth • Boston (MA)

On-site
USD 120,000 - 160,000
Flexible PTO
Competitive pay and equity
Agile, mentor-focused work culture
AI Engineer( Austin based)
AI Engineer( Austin based)

Autonomize • Austin (TX)

On-site
USD 120,000 - 190,000
Real-world impact
Competitive compensation
Employer-paid health, vision, and dent
+2
Senior Software Engineer, Applied AI
Senior Software Engineer, Applied AI

VIVIO Health Inc. • United States

Remote
USD 165,000 - 205,000
Life Insurance
Stock Options
401(k) with Company Match
+3
Director of Engineering: AI, Data & Platform Leader (Hybrid)
Director of Engineering: AI, Data & Platform Leader (Hybrid)

Vumedi Inc. • Oakland (CA)

Hybrid
USD 200,000 - 320,000
Hybrid work schedule
Office in Oakland
Profitable growth and ownership