Principal Machine Learning Engineer

4 days ago


Mountain View, CA, United States Atlassian Full time
Overview

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 Knowledge AI team in US. This is a hands-on technical leadership role where you'll push the boundaries of advanced GenAI modeling, agentic AI systems, rapid prototyping, and applied research. You will turn state-of-the-art techniques into high-impact innovations across Atlassian.

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.

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
  • Provide technical mentorship to team members.
  • Influence design and implementation across retrievers, rankers, orchestration, and serving infrastructure.
  • Collaborate with product and platform teams to turn prototypes into scalable systems.
Shape Innovation Strategy
  • Contribute to the roadmap 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.
Qualifications

Experience
  • 6+ years in ML/AI engineering, 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.
Skills
  • 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.
Education
  • Master's or PhD in Computer Science, Machine Learning, or related fields.
Nice to Have
  • 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 and fostering innovation.

Compensation

At Atlassian, we strive to design equitable, explainable, and competitive compensation programs. To support this goal, the baseline of our range is higher than that of the typical market range, but in turn we expect to hire most candidates near this baseline. Base pay within the range is ultimately determined by a candidate's skills, expertise, or experience. In the United States, we have three geographic pay zones. For this role, our current base pay ranges for new hires in each zone are:

Zone A: USD 232200 - USD 303150

Zone B: USD 209700 - USD 273775

Zone C: USD 193500 - USD 252625

This role may also be eligible for benefits, bonuses, commissions, and equity.

Please visit go.atlassian.com/payzones for more information on which locations are included in each of our geographic pay zones. However, please confirm the zone for your specific location with your recruiter.

Benefits & Perks

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. To learn more, visit go.atlassian.com/perksandbenefits

About Atlassian

At Atlassian, we're motivated by a common goal: to unleash the potential of every team. Our software products help teams all over the planet and our solutions are designed for all types of work. Team collaboration through our tools makes what may be impossible alone, possible together.

We believe that the unique contributions of all Atlassians create our success. To ensure that our products and culture continue to incorporate everyone's perspectives and experience, we never discriminate based on race, religion, national origin, gender identity or expression, sexual orientation, age, or marital, veteran, or disability status. All your information will be kept confidential according to EEO guidelines.

To provide you the best experience, we can support with accommodations or adjustments at any stage of the recruitment process. Simply inform our Recruitment team during your conversation with them.

To learn more about our culture and hiring process, visit go.atlassian.com/crh

  • Mountain View, CA, United States Inworld AI Full time

    About Inworld At Inworld, we believe that the benefits of AI should extend beyond business workflows to the applications and experiences that we enjoy every day. We began by pushing the frontier of lifelike, interactive characters for games and entertainment, pioneering realtime conversational AI at scale. Today, we apply that expertise to provide the...


  • Mountain View, CA, United States Akraya Full time

    Primary Skills: Python (Expert), SQL (Expert), MLOps (Advanced), Cloud Platforms (Intermediate), Data Engineering (Expert), Contract Type: W2 Contract Duration: 6+ Months Location: Mountain View CA (Hybrid, 3Days/Week) Pay Range: $50.00 - $57.00 per hour Job Summary: We are seeking a highly skilled Machine Learning Engineer to join our team in enhancing AI...


  • Mountain View, CA, United States Akraya Full time

    Primary Skills: Python (Expert), SQL (Expert), MLOps (Advanced), Cloud Platforms (Intermediate), Data Engineering (Expert), Contract Type: W2 Contract Duration: 6+ Months Location: Mountain View CA (Hybrid, 3Days/Week) Pay Range: $50.00 - $57.00 per hour Job Summary: We are seeking a highly skilled Machine Learning Engineer to join our team in enhancing AI...


  • Mountain View, CA, United States Akraya Full time

    Primary Skills: Python (Expert), SQL (Expert), MLOps (Advanced), Cloud Platforms (Intermediate), Data Engineering (Expert), Contract Type: W2 Contract Duration: 6+ Months Location: Mountain View CA (Hybrid, 3Days/Week) Pay Range: $50.00 - $57.00 per hour Job Summary: We are seeking a highly skilled Machine Learning Engineer to join our team in enhancing AI...


  • Mountain View, CA, United States Akraya Full time

    Primary Skills: Python (Expert), SQL (Expert), MLOps (Advanced), Cloud Platforms (Intermediate), Data Engineering (Expert), Contract Type: W2 Contract Duration: 6+ Months Location: Mountain View CA (Hybrid, 3Days/Week) Pay Range: $50.00 - $57.00 per hour Job Summary: We are seeking a highly skilled Machine Learning Engineer to join our team in enhancing AI...


  • Mountain View, CA, United States Waymo Full time

    Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver-The World's Most Experienced Driver™-to improve access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo...


  • Mountain View, CA, United States Waymo Full time

    Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver-The World's Most Experienced Driver™-to improve access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo...


  • Mountain View, CA, United States Samsung Electronics Co., Ltd. Full time

    Position Summary Samsung Ads is an advanced advertising technology company in rapid growth that focuses on enabling brands to connect with Samsung TV audiences as they are exposed to digital media by using the industry's most comprehensive data to build the world's smartest advertising platform. Being part of an international company such as Samsung and...


  • Mountain View, CA, United States Samsung Electronics Co., Ltd. Full time

    Position Summary Samsung Ads is an advanced advertising technology company in rapid growth that focuses on enabling brands to connect with Samsung TV audiences as they are exposed to digital media by using the industry's most comprehensive data to build the world's smartest advertising platform. Being part of an international company such as Samsung and...


  • Mountain View, CA, United States Samsung Electronics Co., Ltd. Full time

    Position Summary Samsung Ads is an advanced advertising technology company in rapid growth that focuses on enabling brands to connect with Samsung TV audiences as they are exposed to digital media by using the industry's most comprehensive data to build the world's smartest advertising platform. Being part of an international company such as Samsung and...