Senior AI Knowledge Graph Engineer

WHAT YOU BRING

• Proven experience in developing and deploying to production knowledge graphs in a real business scenario
• Strong hands-on background in developing text-based ML and LLM systems, including prompt context engineering, optimized for production environments
• Strong proficiency in Python, ML / KG frameworks and tools (e.g. Neo4j, NetworkX, Node2Vec, SentenceTransformers) with the ability to select the right tool for the problem at hand
• Ability to apply the appropriate graph algorithms and design choices for problem solving, including relationship and ontology modelling, feature definition, supervised vs unsupervised graph analytics
• Ability to design clean, modular, and testable ML code in a collaborative environment
• Experience with MLOps practices, building production-grade Docker images, and development of AI solutions designed for integration into production systems
• Experience collaborating with software engineers, product managers, and business stakeholders
• Strong communication skills and ability to explain ML concepts and system behavior clearly
• #LI-ONSITE
• WHAT YOU CAN EXPECT FROM US:
• Innovation: We celebrate those who think critically, like a challenge, and aspire to be trailblazers.
• Growth: We give you the space and support to grow along with us and to contribute to something meaningful. We have been named Fortune's Best Workplaces in TechnologyTM, Fortune's Best Workplaces in the Bay AreaTM, and certified as a Great Place to Work®!
• Team: We build each other up and set aside ego for the greater good.
• And because we understand the value of bringing your full and best self to work, we offer a variety of perks to manage a healthy balance, including flexible time off, wellness resources, and company-sponsored team events. Check out purebenefits.com for more information.
• ACCOMMODATIONS AND ACCESSIBILITY:
• Candidates with disabilities may request accommodations for all aspects of our hiring process. For more on this, contact us at TA-Ops@purestorage.com if you’re invited to an interview.
• OUR COMMITMENT TO A STRONG AND INCLUSIVE TEAM:
• We’re forging a future where everyone finds their rightful place and where every voice matters. Where uniqueness isn’t just accepted but embraced. That’s why we are committed to fostering the growth and development of every person, cultivating a sense of community through our Employee Resource Groups and advocating for inclusive leadership.