Senior Machine Learning Systems Engineer

Atlassian; Austin (TX); Hybrid; USD 206,000 - 269,000; Full time

Benefits offered by this job

Health and wellbeing resources

Paid volunteer days

Job summary

At Atlassian, the AI & ML Platform Team is seeking a Senior ML System Engineer to design and optimize large-scale model serving systems. You will own distributed infrastructure—from global KV caches to auto-scaling and low-level GPU kernel optimization—to enable production-grade AI workloads.

You will collaborate with senior ML engineers to fine‑tune open-source LLMs, implement robust CI/CD for inference deployments, and ensure high throughput with low latency under heavy traffic.

Qualifications

• 5+ years of software engineering experience
• 2+ years of system performance optimization experience
• Deep low-level systems programming in C/C++ or Rust
• Experience with large-scale, high-concurrent production serving
• Experience with GPU inference engines (Triton, TensorRT-LLM, vLLM, SGLang) and related optimizations

Responsibilities

• Architect and implement scalable distributed infrastructure for model serving (load balancing, auto-scaling, batch scheduling, global KV cache)
• Optimize latency and throughput of model inference under real production workloads
• Build reliable, high-concurrency serving systems that serve billions of requests reliably
• Benchmark, fine-tune, and accelerate inference engines
• Create robust CI/CD infrastructure for seamless model deployment and inference engine updates
• Partner with senior ML engineers to fine‑tune open-source LLMs and deploy

Skills

C/C++

Rust

Distributed systems

Low-latency optimizations

Performance analysis

Tools

Triton

TensorRT-LLM

vLLM

SGLang

Job description

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.

About The 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 integrating Atlassian products with AI solutions beyond the Atlassian ecosystem.

About the 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, reliable tools that empower Atlassian teams to harness AI. These tools will facilitate the development, deployment, measurement, and operation of AI & ML models and experiences.

Responsibilities

About This Role

As a Senior ML System Engineer On The AI & ML Platform’s Inference Team, You Will Design And Optimize Large-scale Model Serving Systems End-to-end. You Will Own Everything From Distributed Infrastructure (global KV Cache, Continuous Batching, Load Balancing, Auto-scaling) To Deep Low-level Optimizations (GPU Kernels, Quantization, Speculative Decoding).

In This Role, You Are Expected To:

• Architect and implement scalable distributed infrastructure for model serving (load balancing, auto-scaling, batch scheduling, global KV cache).
• Optimize latency and throughput of model inference under real production workloads.
• Build reliable, high-concurrency serving systems that serve billions of requests reliably
• Benchmark, fine-tune, and accelerate inference engines.
• Create robust CI/CD infrastructure for seamless model deployment and inference engine updates.
• Partner with senior ML engineers to fine‑tune open-source LLMs and deploy

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

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: $206,100 - $269,075
• Zone B: $185,490 - $242,168
• Zone C: $171,063 - $223,332

Qualifications

• 5+ years of software engineering experience, 2+ years of system performance optimization experience
• Deep low-level systems programming (C/C++ or Rust)
• Experience with large-scale, high-concurrent production serving.
• Experience with GPU inference engines (vLLM, SGLang, Triton, TensorRT-LLM, etc.).
• Strong background in system optimizations: batching, caching, load balancing, parallelism.

It would be great, but not required if you have

• Low-level inference optimizations: GPU kernels
• Algorithmic inference optimizations: quantization, speculative decoding, distillation
• Experience with testing, benchmarking, and reliability of inference services.
• Experience designing and implementing CI/CD infrastructure for inference.

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.