Senior Machine Learning System Engineer

Atlassian; San Francisco (CA); Hybrid; USD 180,000 - 235,000; Full time

Benefits offered by this job

Health and wellbeing resources

Paid volunteer days

Equity

Job summary

At Atlassian in the United States, the Senior ML System Engineer on the AI & ML Platform team will help build and optimize the core infrastructure that powers ML models and pipelines used by Jira, Confluence, and other products.

You will collaborate with product teams, curate high‑quality ML data, fine‑tune open‑source LLMs or integrate proprietary models, and lead projects from technical design to launch while mentoring junior engineers.

Qualifications

• 5+ years of experience in building Machine Learning and AI infra/platform/system.
• Comprehensive ML lifecycle expertise: developing, deploying, and maintaining end-to-end ML systems.
• Large-scale system design: scalable, fault-tolerant distributed systems for ML.
• MLOps and automation: CI/CD pipelines and automation for continuous training, deployment, and monitoring.

Responsibilities

• Collaborate with product teams to solve complex ML challenges from design to launch.
• Curate high-quality ML datasets and fine-tune open-source LLMs or access proprietary models.
• Access proprietary LLMs and build ML solutions for Atlassian products.
• Lead projects from technical design through to launch and deliver impactful results.
• Mentor junior engineers within the team.

Skills

ML infra

End-to-end ML systems

Large-scale distributed systems

MLOps

Python

PyTorch/TF/JAX

Cloud platforms

Big data processing

Performance optimization

GenAI / LLMs

Job description

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.

Overview

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

About Central AI Org

Our Organization Is Dedicated To Driving AI Innovation Across All Atlassian Products And Platforms. We Aim To Deliver Seamless AI Experiences While Establishing a Robust Atlassian AI Infrastructure For The Future. Our Purpose Is To

• Develop horizontal AI capabilities and infrastructure that can be leveraged across all products.
• Establish a centralized Search, Q&A, and Conversational AI system that integrates seamlessly with all Atlassian products.
• Explore the integration of Atlassian products with AI solutions beyond the Atlassian ecosystem.

About AI & ML Platform Team

Our team’s goal is to build the foundations to democratize AI and Machine Learning for Atlassian’s teams, customers, and ecosystem. We aim to build productive and reliable tools that empower Atlassian teams to harness the power of AI. These tools will facilitate the development, deployment, measurement, and operation of AI & ML experiences.

Our tools are designed to integrate seamlessly with other Atlassian platforms, including the Atlassian Data Platform. This integration enables teams to efficiently and swiftly incorporate AI and ML capabilities into their workflows while strictly adhering to all security and data usage policies. Our primary goal is to deliver a smooth and hassle‑free experience for Atlassian users, empowering them to harness the potential of AI and ML without any complications.

About This Role

As a Senior ML System Engineer on the AI & ML Platform team, you will play a pivotal role in developing and refining the core infrastructure that empowers all Atlassian software engineers, ML engineers, and data scientists to create, train, evaluate, deploy, and manage Machine Learning models and pipelines.

You will collaborate closely with product teams, such as Jira and Confluence, to solve their specific challenges in building ML solutions. This may involve curating high‑quality ML datasets, fine‑tuning open‑sourced Large Language Models (LLMs), or accessing proprietary LLMs. Your expertise in both ML and software development expertise will be instrumental in overcoming challenging problems and navigating complex infrastructure and architectural issues.

This position offers you the chance to lead projects from the technical design phase all the way to launch. You will partner with various teams and internal stakeholders to achieve impactful results.

In This Role, You’ll Get The Chance To

• Collaborate with your teammates to solve complex problems, from technical design to launch.
• Deliver cutting‑edge solutions that are used by other Atlassian teams and products to build AI features that reach millions of customers.
• Deliver code reviews, documentation & bug fixes within a strong engineering culture
• Partner across engineering teams to take on company‑wide initiatives spanning multiple projects.
• Mentor junior members of the team.

On your first day, we’ll expect you to have

• 5+ years of experience in building Machine Learning and AI infra/platform/system
• Comprehensive ML lifecycle expertise: proven experience developing, deploying, and maintaining end‑to‑end ML systems, from data engineering to model serving and monitoring.
• Large‑scale system design: Extensive experience designing and building scalable, fault‑tolerant, and high‑performance distributed systems for machine learning.
• MLOps and automation: Deep experience implementing MLOps, CI/CD pipelines, and automation for continuous training, deployment, and monitoring of ML models.

It would be great, but not required if you have

• Proficiency with frameworks and languages: Expert‑level proficiency in Python and ML frameworks like PyTorch, TensorFlow, or JAX. Familiarity with other languages like Go, Java, or Scala is also beneficial.
• Cloud infrastructure: Hands‑on expertise with major cloud platforms such as AWS, GCP, or Azure, including their specific AI/ML services and compute resources like GPUs.
• Big data processing: Experience with distributed computing frameworks for large‑scale data processing, such as Spark, Ray, or Dask.
• Performance optimization: A demonstrated ability to diagnose and solve complex performance and optimization problems for ML models and infrastructure.
• Generative AI systems: Experience with GenAI frameworks and tools, including developing and fine‑tuning large language models (LLMs) and building retrieval‑augmented generation (RAG) systems.

Compensation

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 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.

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

Pay Ranges

Role

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: $180,000 - $235,000
• Zone B: $162,000 - $211,500
• Zone C: $149,400 - $195,050

Qualifications

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.

In line with local law, identity verification (which may include use of biometric data) is a condition of employment with Atlassian for employment fraud purposes.