Principal Machine Learning Engineer
About The Position
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
- 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.
- health and wellbeing resources
- paid volunteer days
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Principal
Number of Employees
5,001-10,000 employees
