Machine Learning Engineer (Agentic Search & Knowledge Graphs)

Company: Salesforce, Inc.
Location: San Francisco
Posted on: March 25, 2025

Job Description:

Machine Learning Architect - Search & Knowledge GraphsAbout SalesforceWe're Salesforce, the Customer Company, inspiring the future of business with AI+ Data +CRM. Leading with our core values, we help companies across every industry blaze new trails and connect with customers in a whole new way. And, we empower you to be a Trailblazer, too - driving your performance and career growth, charting new paths, and improving the state of the world. If you believe in business as the greatest platform for change and in companies doing well and doing good - you've come to the right place.The RoleSalesforce is seeking a visionary Machine Learning Architect to lead advancements in intelligent Search and Knowledge Graph solutions within our Einstein Foundation team. This role is pivotal in redefining how we enable innovative, knowledge-driven experiences across Salesforce's next-gen AI products. As an authority in Search, Knowledge Graphs, and Large Language Models (LLMs), you will drive the evolution of Salesforce's AI systems with innovative retrieval, representation, and context expansion technologies that serve millions of users globally.What You'll Do:

• Lead the Architecture of Sophisticated Search & Knowledge Graph SolutionsArchitect and implement end-to-end, large scale search and retrieval solutions that demonstrate Knowledge Graphs and are optimized for high-performance, multi-tenant environments.
• Develop Intelligent Retrieval PipelinesInnovate hybrid retrieval pipelines combining semantic, vector, and symbolic search to improve contextual relevance, speed, and accuracy in knowledge driven AI applications.
• Optimize and Automate Search SystemsEnhance system efficiency through automation in demand forecasting, configuration, and proactive monitoring, driving real-time search optimization.
• Collaborate Across Teams for AI Driven Product InnovationWork closely with multi-functional teams, including Product Managers, Knowledge Engineers, and ML Researchers, to assemble requirements and translate them into scalable, innovative search and retrieval solutions.
• Pioneer Search and Knowledge Graph InnovationsGuide discussions on emerging technologies and advancements in vector search, graph embeddings, and knowledge augmented retrieval, valuing continuous innovation.Required Skills:
  • 15+ years in Machine Learning & Search SystemsExtensive experience with large-scale search, Machine Learning, and knowledge driven systems, specifically focused on integrating Knowledge Graphs, search optimization, and sophisticated retrieval techniques.
  • Expertise in Semantic and Vector-Based SearchDeep knowledge of vector databases (e.g., FAISS, Pinecone, Milvus), approximate nearest neighbor (ANN) search algorithms, and embedding techniques to power high relevance search systems.
  • Strong Background in NLP & LLMsExperience with natural language processing (NLP), prompt engineering, and applying LLMs to enhance knowledge based search and retrieval in enterprise contexts.
  • Sophisticated Knowledge Graph SkillsProficiency in graph databases (e.g., Neo4j, Amazon Neptune), graph embedding, and linking techniques to enable rich contextual search and high dimensional graph-based retrieval.
  • Proficiency in Distributed Systems & ML FrameworksAuthority understanding of distributed systems, data streaming (e.g., Kafka, Spark), and Machine Learning frameworks (TensorFlow, PyTorch) to support realtime, resilient AI applications.
  • Programming Mastery in Python & Graph Based FrameworksStrong programming skills in Python, with expertise in machine learning and graph-based frameworks to facilitate scalable, high-performance AI solutions.Preferred Search & Knowledge Graph-Specific Skills:
    • Experience with Multi-Stage Retrieval PipelinesHands-on experience in designing and optimizing multi-stage retrieval workflows that balance precision, recall, and relevance at scale.
    • In-Depth Knowledge of Re-Ranking & Retrieval OptimizationExpertise in retrieval-specific optimizations, including re-ranking, hybrid search, and knowledge augmented retrieval, to improve relevance in enterprise-scale systems.
    • Graph Embedding & Contextual Retrieval ExpertiseConfirmed skills in graph based search, context expansion techniques, and Knowledge Graph integration to enhance retrieval depth and accuracy.
    • Knowledge Graph Curation & Ontology ManagementExperience in Knowledge Graph curation, schema design, and ontology management, ensuring efficient and adaptable knowledge driven search solutions.
    • Familiarity with Feedback Loops and Fine-TuningKnowledge of incorporating user feedback and relevance signals to fine-tune contextual embeddings and improve Search and Knowledge Graph system performance.Additional Preferred Skills:
      • Broad ML Experience with Diverse ApproachesStrong foundation in diverse ML techniques, from neural networks to probabilistic models, adaptable for Search and Knowledge Graph-centric AI use cases.
      • Exceptional Communication and Collaboration SkillsOutstanding written and verbal communication abilities, with confirmed expertise in collaborating across engineering, research, and product teams.If you're an industry leader passionate about Search, Knowledge Graphs, and innovative AI, and eager to make an impact at the world's #1 CRM company, we'd love to meet you!
        #J-18808-Ljbffr

Keywords: Salesforce, Inc., San Francisco , Machine Learning Engineer (Agentic Search & Knowledge Graphs), Engineering , San Francisco, California