Senior Machine Learning Systems Engineer

Overview

Atlassian drives its mission to unlock every team's full potential through innovative collaboration tools like Jira, Confluence, and Bitbucket. These products enable teams across industries to work together effectively and achieve excellence — impacting endeavors from NASA's Mars missions to Cochlear's hearing innovations.

About the Central AI Organization

This group spearheads AI advancements across Atlassian's portfolio by creating foundational AI capabilities and infrastructure. Their focus includes building unified Search, Q&A, and Conversational AI systems integrated across Atlassian products as well as exploring AI partnerships beyond.

AI & ML Platform Team Purpose

The team develops the core infrastructure and tools to democratize AI and machine learning usage internally and externally. Their solutions seamlessly blend with Atlassian’s platforms to help teams rapidly adopt AI/ML capabilities while maintaining strict security and data policies.

Working Environment

Atlassians enjoy flexible work locations including offices, remote, or hybrid setups in countries where Atlassian holds legal entities.

Key Responsibilities

• Design and implement foundational systems that facilitate ML model creation, training, evaluation, deployment, and maintenance.
• Collaborate directly with product teams such as Jira and Confluence to tailor solutions for complex ML challenges, including dataset preparation and large language model fine-tuning.
• Lead projects end-to-end from architecture design through delivery, engaging stakeholders across the company.
• Conduct code reviews, write documentation, address bugs, and uphold a strong engineering culture.
• Mentor junior engineers and support cross-team initiatives with wide-reaching impact.

Candidate Qualifications

• Over 5 years of experience building ML systems, platforms, or infrastructure.
• Expertise through the full ML lifecycle covering data engineering, model development, deployment, and monitoring.
• Proven ability designing scalable, performant, and resilient distributed ML systems.
• Strong coding skills in Python with familiarity in ML frameworks such as PyTorch, TensorFlow, or JAX; experience in Go, Java, or Scala is advantageous.
• Hands-on experience implementing MLOps practices including CI/CD pipelines for continuous model training and monitoring.

Preferred Additional Skills

• Practical knowledge of major cloud platforms AWS, GCP, or Azure and their AI/ML offerings, including GPU compute resources.
• Familiarity with big data frameworks like Spark, Ray, or Dask for large-scale data processing.
• Ability to diagnose and optimize ML model and infrastructure performance.
• Experience with generative AI, including development and customization of large language models and retrieval-augmented generation systems.

Benefits and Inclusion

Atlassian provides comprehensive support for your well-being and family including health resources and paid volunteer opportunities. The company values diversity and inclusivity, ensuring a discrimination-free environment respecting all backgrounds and identities. Reasonable accommodations are available throughout recruitment and employment. Identity verification may include biometrics as required by law for employment fraud prevention.