Staff ML Engineer: Knowledge Graphs for AI-Driven Data

Prophecy; Palo Alto (CA); Hybrid; USD 250,000 - 350,000; Full time

Benefits offered by this job

Health coverage

Wellness stipend

Flexible PTO

Professional development

Long-term disability

Dental & vision

Job summary

Prophecy is seeking a Staff Machine Learning Engineer specializing in Knowledge Graphs in San Francisco, hybrid. The role leads end-to-end KG projects, builds scalable graph systems, and integrates AI agents with the data platform.

You will shape enterprise data prep and insights with cutting-edge tech. The position offers a base salary, equity, and comprehensive benefits, and requires strong Python, AI, cloud, and graph database skills.

Qualifications

• Knowledge Graphs: entity extraction, linkage, ontology design.
• KG integration with AI systems for retrieval and context.
• Tech lead experience delivering end-to-end KG projects.
• Proficient in Python and REST APIs; cloud-native microservices.
• Experience with Go, AWS, Kubernetes, Java, Scala.
• Hands-on with graph databases like Neo4j.

Responsibilities

• Own end-to-end KG projects from design to deployment.
• Lead cross-functional teams and drive delivery.
• Design and scale knowledge graph systems powering data prep and analysis.
• Collaborate with AI agents to integrate enterprise data infrastructure.

Skills

Knowledge graphs

AI integration

Technical leadership

Python

REST APIs

Cloud microservices

Go

AWS

Kubernetes

Java

Scala

Tools

Neo4j

PyTorch

Python

SQL

Neo4j/Graph DBs

Job description

Prophecy is seeking a Staff Machine Learning Engineer specializing in Knowledge Graphs in San Francisco, hybrid. The role leads end-to-end KG projects, builds scalable graph systems, and integrates AI agents with the data platform.

You will shape enterprise data prep and insights with cutting-edge tech. The position offers a base salary, equity, and comprehensive benefits, and requires strong Python, AI, cloud, and graph database skills.