Principal AI/ML Architect: Discovery & Recommendations

Vumedi; Oakland (CA); Hybrid; USD 180,000 - 250,000; Full time

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