Principal Machine Learning Engineer

AtlassianSeattle, WA
14h$193,500 - $303,150Hybrid

About The Position

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. As part of that mission, we're investing deeply in Generative AI — pioneering advanced modeling and rapid innovations that accelerate how teams work, discover, and create. We’re seeking a Principal Machine Learning Engineer (P60) to join our new GenAI Modeling & Jira Search team. This is a hands-on technical leadership role where you’ll push the boundaries of advanced GenAI modeling, rapid prototyping, and applied research—turning state-of-the-art techniques into high-impact innovations across Atlassian.

Requirements

  • Extensive experience in ML/AI engineering (generally 6+ years), with deep expertise in LLMs, NLP, or search.
  • Proven track record of building and shipping ML-powered or GenAI-integrated products.
  • Hands-on experience with retrieval-augmented generation (RAG), embeddings, hybrid retrieval, or agentic workflows.
  • Strong background in scaling ML systems from prototype to production.
  • Expert-level knowledge of GenAI ecosystems (OpenAI, Anthropic, Mistral, OSS) and fine-tuning/serving frameworks.
  • Deep technical skills across modeling, infrastructure, and orchestration (LangChain, LlamaIndex, vector DBs).
  • Strong prototyping ability, balancing speed with rigor in experimentation.
  • Ability to set technical direction and inspire peers with thought leadership.
  • Master’s or PhD in Computer Science, Machine Learning, or related field preferred—or equivalent industry experience.

Nice To Haves

  • Experience with multimodal models, knowledge graphs, or semantic embeddings.
  • Familiarity with evaluation metrics for search/GenAI (NDCG, groundedness, human preference modeling).
  • Contributions to open-source ML frameworks or GenAI research communities.
  • Passion for mentoring, fostering innovation, and shaping a new AI hub in Singapore.

Responsibilities

  • Advance GenAI Modeling with Knowledge Design and build novel approaches for knowledge-grounded LLMs, embeddings, and multimodal modeling.
  • Explore fine-tuning, post-training, and retrieval-augmented strategies to adapt foundation models to Atlassian’s data.
  • Lead technical explorations that expand the frontier of what’s possible in applied generative modeling.
  • Prototype & Experiment Rapidly Drive POC experimentation, from agentic workflows to novel GenAI-powered assistants.
  • Translate research ideas into fast, working prototypes that demonstrate user value.
  • Define and implement evaluation methods (groundedness, hallucination detection, human-in-the-loop) to measure success.
  • Provide Technical Leadership Act as the technical anchor of a small tiger team (manager + ICs), mentoring engineers and scientists.
  • Influence design and implementation across retrievers, rankers, orchestration, and serving infrastructure.
  • Collaborate with product and platform teams to transition prototypes into scalable systems.
  • Shape Innovation Strategy Contribute to the roadmap for the Forge as a global hub for advanced GenAI modeling and applied research.
  • Partner with peers across Rovo, Jira Search, Knowledge AI, and Platform to drive cross-team impact.
  • Champion responsible AI, groundedness, and data quality in all innovations.

Benefits

  • Atlassians can choose where they work – whether in an office, from home, or a combination of the two.
  • Atlassian offers a wide range of perks and benefits designed to support you, your family and to help you engage with your local community.
  • Our offerings include health and wellbeing resources, paid volunteer days, and so much more.
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service