Drimva

Atlassian

Principal Machine Learning Engineer

$196K–$309K

Seattle, United States +3 · Principal · Full-time · Posted Oct 28, 2026

Location
Seattle, United States
Remote, Remote
Mountain View, United States or Remote
San Francisco, United States
Experience
8+ years in ML/AI engineering.
Education
Master's or PhD in Computer Science, Machine Learning, Information Retrieval, or related field preferred — or equivalent industry experience.

About the job

Leads development of personal work-context knowledge graphs and delivers permission-safe graph context to Rovo Chat, APIs, CLI, and agents.

What you'll do

  • Design graph inference pipelines, schemas, permissions, and evaluation frameworks for inferred work context.
  • Integrate graph context into Rovo Chat and build low-latency APIs and CLI experiences for real-time access.
  • Set technical direction across teams, mentor engineers, and promote responsible, privacy-safe AI and data quality.

What they ask for

  • 8+ years in ML/AI engineering, with deep expertise in knowledge graphs, graph neural networks, or entity/relationship extraction.
  • Production-scale track record building and shipping ML-powered graph inference or knowledge representation systems.
  • Hands-on experience with graph databases, graph query languages, or large-scale graph processing frameworks.
  • Experience shipping end-to-end ML features and understanding LLM orchestration, RAG, context injection, and ML evaluation.

Nice to have

  • Experience with enterprise knowledge graphs, semantic embeddings, or ontology design at scale.
  • Familiarity with permission-aware systems and privacy-by-design; background in collaboration analytics, social network analysis, or user activity modeling.

Pay

$236,700 - $309,025 USD base salary; $213,030 - $278,123 USD base salary; $196,461 - $256,491 USD base salary

Tools

Neo4j, Neptune, Cypher, SPARQL, GraphX, DGL, PyG, Rovo Chat

AI focus

Builds knowledge-graph inference and personal work-context systems; uses graph-derived context to improve LLM grounding and relevance in Rovo Chat and enable real-time agent queries.

Model-written summary · the original posting is authoritative

First seen Oct 28, 2026 · Last verified Nov 5, 2026 (UTC)