Principal ML Engineer: Knowledge Graphs & RAG Context

Atlassian; San Francisco (CA); On-site; USD 237,000 - 309,000; Full time

Job summary

At Atlassian, we seek a Principal Machine Learning Engineer in San Francisco to lead knowledge graph initiatives that power a personal working environment context for Rovo Chat and the Teamwork Graph CLI. You will design graphs, schemas, and APIs, and mentor teams across Knowledge AI and product groups.

Candidates should have 8+ years in ML/AI with graph tech, and a track record shipping end-to-end ML features at production scale.

Qualifications

• 8+ years in ML/AI eng with knowledge graphs or similar.
• Production ML features: data pipeline to deployment.
• Experience with graph databases and query languages.
• Experience shipping end-to-end ML systems at scale.

Responsibilities

• Design and build Personal Work Context Graphs and schemas.
• Improve Rovo Chat with graph context and relevance.
• Deliver low-latency, permission-safe APIs and CLI.
• Provide technical leadership across Knowledge AI and Graph teams.

Skills

Knowledge graphs

Graph neural networks

Entity extraction

Production ML

Graph databases

Cypher/SPARQL

LLM orchestration

RAG

Context injection

Education

Master's or PhD in CS/ML

Tools

Neo4j

Neptune

GraphX

DGL

PyG

Job description

At Atlassian, we seek a Principal Machine Learning Engineer in San Francisco to lead knowledge graph initiatives that power a personal working environment context for Rovo Chat and the Teamwork Graph CLI. You will design graphs, schemas, and APIs, and mentor teams across Knowledge AI and product groups.

Candidates should have 8+ years in ML/AI with graph tech, and a track record shipping end-to-end ML features at production scale.