GenAI ML Systems Architect & Lead

Atlassian; Washington, Northern (District of Columbia, KY); Hybrid; USD 237,000 - 309,000; Full time

Benefits offered by this job

Bonuses

Equity

Commissions

Health benefits

Job summary

At Atlassian, we’re advancing Generative AI to empower teams worldwide. We seek a Principal ML Systems Engineer to design and scale GenAI systems, tackle training, inference, and knowledge-grounded pipelines, and partner with cross‑functional teams to ship production‑ready innovations.

You will influence architecture, optimize performance, and help establish best practices for deployment, monitoring, and evaluation across cloud environments and modern containerized infrastructure.

Qualifications

• 6+ years in ML systems engineering or related roles.
• Proven ability to scale ML-powered services in production.
• Experience with large-scale training and inference pipelines.
• Proficiency in Python and ML frameworks (PyTorch, TensorFlow).
• Experience with vector databases and orchestration tools.
• Strong coding skills and performance-focused mindset.
• Familiarity with AWS, GCP, and Azure cloud environments.

Responsibilities

• Design and build scalable ML systems for training, fine-tuning, and serving LLMs.
• Develop retrieval, hybrid search, and RAG pipelines with knowledge grounding.
• Create tools and infra for rapid experimentation and deployment.
• Partner with scientists to productionize prototypes.
• Evolve POC systems into reliable services.
• Optimize latency, throughput, and resource use for GenAI workloads.
• Collaborate with ML engineers, backend teams and product to ship end-to-end innovations.
• Establish model deployment, monitoring, and evaluation best practices.
• Position Atlassian as a hub for GenAI systems innovation.

Skills

ML systems engineering

Backend engineering

Model training

Python

PyTorch

TensorFlow

Hugging Face

Weaviate

Pinecone

FAISS

LangChain

LlamaIndex

AWS

GCP

Azure

Kubernetes

Docker

Education

Bachelor’s or Master’s in Computer Science, ML, or related field

Tools

PyTorch

TensorFlow

Hugging Face

Weaviate

Pinecone

FAISS

LangChain

LlamaIndex

Kubernetes

Docker

Job description

At Atlassian, we’re advancing Generative AI to empower teams worldwide. We seek a Principal ML Systems Engineer to design and scale GenAI systems, tackle training, inference, and knowledge-grounded pipelines, and partner with cross‑functional teams to ship production‑ready innovations.

You will influence architecture, optimize performance, and help establish best practices for deployment, monitoring, and evaluation across cloud environments and modern containerized infrastructure.