Machine Learning System Engineer

Job description

Overview

Working at Atlassian.

Atlassians can choose where they work – in an office, from home, or a combination of the two.

We hire in any country where we have a legal entity.

Interviews and onboarding are conducted virtually as part of being a distributed-first company.

About Atlassian

Atlassian's mission is to unleash the potential of every team.

Our products, such as Jira, Confluence, and Bitbucket, support teamwork across diverse teams and processes to help teams succeed.

We power revolutions across many industries by enabling teams to collaborate effectively and achieve their highest potential.

About Central AI Org

Our organization drives AI innovation across Atlassian products and platforms with the aim of delivering seamless AI experiences and a robust Atlassian AI infrastructure.

Our purpose is to:

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

About AI & ML Platform Team

We build foundations to democratize AI and Machine Learning for Atlassian’s teams, customers, and ecosystem.

Our tools enable development, deployment, measurement, and operation of AI & ML experiences and integrate with the Atlassian Data Platform while respecting security and data usage policies.

The goal is a smooth, hassle-free experience for Atlassian users to harness AI and ML.

About This Role

As a Machine Learning System Engineer on the AI & ML Platform team, you will help develop and refine the core infrastructure that empowers Atlassian software engineers, ML engineers, and data scientists to create, train, evaluate, deploy, and manage ML models and pipelines.

You will collaborate with product teams such as Jira and Confluence to solve challenges in building ML solutions, which may involve curating ML datasets, fine-tuning open-sourced LLMs, or accessing proprietary LLMs. Your ML and software development expertise will help overcome infrastructure and architectural challenges.

This role offers opportunities to lead projects from design to launch and to partner with various teams and stakeholders to achieve impact.

In This Role, You'll Get The Chance To

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

On Your First Day, We’ll Expect You To Have

• Extensive experience in building Machine Learning and AI infra/platform/system (generally 3+ years).
  
  
• Comprehensive ML lifecycle expertise: developing, deploying, and maintaining end-to-end ML systems, from data engineering to model serving and monitoring.
  
  
• Large-scale system design: designing and building scalable, fault-tolerant, high-performance distributed systems for ML.
  
  
• Proficiency with Python and ML frameworks like PyTorch, TensorFlow, or JAX.
  
  Familiarity with Go, Java, or Scala is beneficial.
  
  
• MLOps and automation: experience implementing MLOps, CI/CD pipelines, and automation for continuous training, deployment, and monitoring of ML models.
  
  

Nice-to-Have (Not Required)

• Cloud infrastructure: experience with AWS, GCP, or Azure and their AI/ML services and GPUs.
• Big data processing: experience with Spark, Ray, or Dask.
  
  
• Performance optimization: ability to diagnose and solve ML performance issues.
  
  
• Generative AI systems: experience with GenAI frameworks, LLMs, and retrieval-augmented generation (RAG).
  
  

Qualifications

Benefits & Perks

Atlassian offers a wide range of perks and benefits designed to support you and your family and to help you engage with your local community.

Our offerings include health and wellbeing resources, paid volunteer days, and more.

To learn more, visit go.atlassian.com/perksandbenefits .

EEO & Recruitment

At Atlassian, we never discriminate based on race, religion, national origin, gender identity or expression, sexual orientation, age, marital status, veteran status, or disability status.

All your information will be kept confidential according to EEO guidelines.

We can provide accommodations or adjustments at any stage of the recruitment process; inform our Recruitment team during your conversation.

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

Seniority level

• Mid-Senior level

Employment type

• Full-time

Job function

• Engineering and Information Technology

Industries

• Software Development

Referrals increase your chances of interviewing at Atlassian.

Sign in to set job alerts for “Machine Learning Engineer” roles.

Required Skill Profession

Electrical & Energy Engineering