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
Develop and implement advanced machine learning algorithms and models for Atlassian products. Design system and model architectures, conduct experiments and evaluations, collaborate with product, engineering, and analytics teams, and mentor emerging machine learning engineers. The role also contributes to shared AI infrastructure and capabilities supporting search, knowledge discovery, and conversational experiences across the company.
Top Skills:
Ai InfrastructureArtificial IntelligenceData PipelinesMachine LearningMachine Learning AlgorithmsMachine Learning FrameworksModel EvaluationModel Training
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
What you need to know about the Montreal Tech Scene
With roots dating back to 1642, Montreal is often recognized for its French-inspired architecture and cobblestone streets lined with traditional shops and cafés. But what truly sets the city apart is how it blends its rich tradition with a modern edge, reflected in its evolving skyline and fast-growing tech industry. According to economic promotion agency Montréal International, the city ranks among the top in North America to invest in artificial intelligence, making it le spot idéal for job seekers who want the best of both worlds.
Key Facts About Montreal Tech
- Number of Tech Workers: 255,000+ (2024, Tourisme Montréal)
- Major Tech Employers: SAP, Google, Microsoft, Cisco
- Key Industries: Artificial intelligence, machine learning, cybersecurity, cloud computing, web development
- Funding Landscape: $1.47 billion in venture capital funding in 2024 (BetaKit)
- Notable Investors: CIBC Innovation Banking, BDC Capital, Investissement Québec, Fonds de solidarité FTQ
- Research Centers and Universities: McGill University, Université de Montréal, Concordia University, Mila Quebec, ÉTS Montréal
