Lead the design and delivery of knowledge graph and machine learning systems that infer personal work context from connected tools. Build graph inference pipelines, schemas, permission-aware APIs, Rovo Chat integrations, and MCP-compatible CLI experiences. Establish evaluation frameworks, improve retrieval and grounding, provide technical direction across teams, mentor engineers, and ensure responsible, privacy-safe AI systems operate reliably at production scale.
Engineering | Seattle, United States | Remote, Remote | Mountain View, United States or Remote | San Francisco, United States
Working at Atlassian
Atlassians can choose where they work - whether in an office, from home, or a combination of the two. That way, Atlassians have more control over supporting their family, personal goals, and other priorities. We can hire people in any country where we have a legal entity.
At Atlassian, we're on a mission to unleash the potential of every team. Central to that mission is the Teamwork Graph (TWG) - Atlassian's real-time, permissions-aware knowledge graph that unifies people, teams, projects, content, and activities across Atlassian and connected third-party tools. We believe the next frontier of AI-powered teamwork is personal working environment context: knowing who you collaborate with, what you're actively working on, and which documents matter right now - so that Rovo Chat, agents, and the TWG CLI can deliver answers that are precise, relevant, and actionable.
We're seeking a Principal Machine Learning Engineer (P60) to lead and design knowledge graph projects that build this personal working environment context layer and serve it at scale through Rovo Chat and the Teamwork Graph CLI.
What You'll Do
Build Personal Work Context Graphs
Improve Rovo Chat with Graph Context
Deliver Context Through Graph APIs & CLI
Lead Across Teams
At Atlassian, we strive to design equitable, explainable, and competitive compensation programs. We follow consistent hiring practices and account for each candidate's skills, knowledge, and experience when setting base pay within the range.
Please visit
Working at Atlassian
Atlassians can choose where they work - whether in an office, from home, or a combination of the two. That way, Atlassians have more control over supporting their family, personal goals, and other priorities. We can hire people in any country where we have a legal entity.
At Atlassian, we're on a mission to unleash the potential of every team. Central to that mission is the Teamwork Graph (TWG) - Atlassian's real-time, permissions-aware knowledge graph that unifies people, teams, projects, content, and activities across Atlassian and connected third-party tools. We believe the next frontier of AI-powered teamwork is personal working environment context: knowing who you collaborate with, what you're actively working on, and which documents matter right now - so that Rovo Chat, agents, and the TWG CLI can deliver answers that are precise, relevant, and actionable.
We're seeking a Principal Machine Learning Engineer (P60) to lead and design knowledge graph projects that build this personal working environment context layer and serve it at scale through Rovo Chat and the Teamwork Graph CLI.
What You'll Do
Build Personal Work Context Graphs
- Design graph inference pipelines that surface collaborators, active work, documents, and projects from connected tools.
- Define schemas, permissions, and evaluation frameworks for reliable inferred context.
Improve Rovo Chat with Graph Context
- Integrate personal context into Rovo Chat to improve relevance, groundedness, and efficiency.
- Build and measure context-selection strategies with the Rovo Chat team.
Deliver Context Through Graph APIs & CLI
- Build low-latency, permission-safe APIs and CLI experiences for personal work context.
- Enable MCP-compatible agents to query a user's work environment in real time.
Lead Across Teams
- Provide technical leadership across Knowledge AI, Teamwork Graph, and product teams.
- Mentor engineers and champion responsible, privacy-safe AI and data quality.
At Atlassian, we strive to design equitable, explainable, and competitive compensation programs. We follow consistent hiring practices and account for each candidate's skills, knowledge, and experience when setting base pay within the range.
Please visit
Similar Jobs at Atlassian
Cloud • Information Technology • Productivity • Security • Software • App development • Automation
Principal Machine Learning Engineer responsible for setting technical direction and building production-grade AI systems for Confluence. The role spans model evaluation, retrieval, ranking, prompt and workflow design, experimentation, reliability, and customer-facing AI product development. Responsibilities include making architectural decisions, partnering across engineering and product teams, mentoring engineers, identifying model and product failure modes, and delivering scalable AI experiences across content creation, editing, discovery, recommendations, and multimodal interaction.
Top Skills:
Ai Systems InfrastructureArtificial IntelligenceAutomated EvaluationExperimentationMachine LearningMultimodal AiPrompt EngineeringRankingRetrieval
Cloud • Information Technology • Productivity • Security • Software • App development • Automation
Lead machine learning architecture and technical direction across multiple quarters while remaining involved in experimentation, quality, and production systems. Work across model post-training, personalization and recommendation, and agentic experiences. Define solutions for ambiguous problems, align cross-functional partners, drive execution, and safely bring ML systems into production.
Top Skills:
Ai AgentsMachine LearningRecommendation Systems
Cloud • Information Technology • Productivity • Security • Software • App development • Automation
Lead the technical direction for Atlassian’s Search Serving platform, building globally scalable, low-latency agentic search systems. Own architecture for retrieval, indexing, query processing, and ML inference while improving quality, reliability, scalability, operability, and cost. Establish production ML practices for evaluation, experimentation, observability, rollouts, and incident response. Optimize inference for embedding, retrieval, and ranking models, and lead cross-team initiatives while mentoring senior engineers and aligning technical leaders.
Top Skills:
CachingDistillationDistributed SystemsEmbedding ModelsHybrid RetrievalInformation RetrievalMachine LearningMl InferenceModel OptimizationQuantizationRanking ModelsSearch InfrastructureSemantic SearchShardingVector Search
What you need to know about the Delhi Tech Scene
Delhi, India's capital city, is a place where tradition and progress co-exist. While Old Delhi is known for its rich history and bustling markets, New Delhi is defined by its modern architecture. It's clear the region places a strong emphasis on preserving its cultural heritage while embracing technological advancements, particularly in artificial intelligence, which plays a central role in shaping the city's tech landscape, fueled by investments in research and development.
