Atlassian

Principal Machine Learning Engineer

Atlassian
Apply
1 day ago
Remote, United States +3 moreStaff+
H1B sponsor

Base Salary

$196k - $309k/yr

Responsibilities

  • 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.
  • Integrate personal work context into Rovo Chat and build context-selection strategies that improve relevance, groundedness, and efficiency.
  • Build low-latency, permission-safe APIs and CLI experiences and enable MCP-compatible agents to query users' work environments in real time.
  • Lead architecture and technical direction across Knowledge AI, Teamwork Graph, and product teams.
  • Mentor engineers and champion responsible, privacy-safe AI and data quality.

Requirements

  • 8+ years of experience in ML/AI engineering, including deep expertise in knowledge graphs, graph neural networks, or entity and relationship extraction.
  • A proven track record of building and shipping ML-powered graph inference or knowledge representation systems at production scale.
  • Hands-on experience with graph databases, graph query languages, or large-scale graph processing frameworks.
  • Experience shipping end-to-end ML features from data pipelines and model training through serving, monitoring, and iteration.
  • Strong understanding of LLM orchestration, retrieval-augmented generation, and context injection.
  • Experience designing inference pipelines from heterogeneous activity signals such as work items, documents, projects, and code changes.
  • Proficiency in offline precision/recall benchmarks, online A/B testing, and human evaluation for ML systems.
  • Ability to set technical direction across teams, drive architecture decisions, and communicate tradeoffs to engineering and product leadership.
  • A Master's or PhD in Computer Science, Machine Learning, Information Retrieval, or a related field is preferred, or equivalent industry experience.
  • Preferred experience includes enterprise knowledge graphs, semantic embeddings, ontology design, permission-aware data systems, privacy-by-design, collaboration analytics, social network analysis, or user activity modeling.

Benefits

  • Flexible work arrangement allowing employees to work from an office, from home, or in a combination of both.
  • The role may be hired in any country where Atlassian has a legal entity.
  • Benefits may include health and wellbeing resources and paid volunteer days.
  • The role may also be eligible for bonuses, commissions, equity, and other benefits.

Tech Stack

Neo4j

Categories

Atlassian

About Atlassian

10,000+ employees

Atlassian powers the collaboration that helps teams accomplish what would otherwise be impossible alone. From space missions and motor racing to bugs in code and IT requests, no task is too large or too small with the right team, the right tools, and the right practices. Over 300,000 global companies and 80% of the Fortune 500 rely on Atlassian’s software, like Jira, Confluence, Loom, and Trello, to help their teams work better together and deliver quality results on time. With our 300,000+ customers and team of 10,000+ Atlassians, we are building the next generation of team collaboration and productivity software. We believe the power of teams has the potential to change the world — one that is more open, authentic, and inclusive.

Contact me