About the Role
As a Principal 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 to solve their specific challenges in building ML solutions, and lead projects from technical design to launch.
Responsibilities
- 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 & 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.
Requirements
- 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.
- 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.
- Deep experience implementing MLOps, CI/CD pipelines, and automation for continuous training, deployment, and monitoring of ML models.
- Hands-on expertise with major cloud platforms such as AWS, GCP, or Azure, including their specific AI/ML services and compute resources like GPUs.
- Experience with distributed computing frameworks for large-scale data processing, such as Spark, Ray, or Dask.
- A demonstrated ability to diagnose and solve complex performance and optimization problems for ML models and infrastructure.
- Experience with GenAI frameworks and tools, including developing and fine-tuning large language models (LLMs) and building retrieval-augmented generation (RAG) systems.
Skills
- Python
- PyTorch
- TensorFlow
- JAX
- Go
- Java
- Scala
- MLOps
- CI/CD
- AWS
- GCP
- Azure
- Spark
- Ray
- Dask
- Generative AI
- LLMs
- RAG
Location
- Any country where Atlassian has a legal entity
Work Type
- Office, home, or a combination of the two
Experience Level
- Principal Engineer
- 5+ years of experience
Salary/Compensations
- Zone A: $209,700 - $273,775
- Zone B: $188,730 - $246,398
- Zone C: $174,051 - $227,233
Benefits
- Health and wellbeing resources
- Paid volunteer days
- Benefits, bonuses, commissions, and equity may be available
- Accommodations or adjustments at any stage of the recruitment process
About the Company
- Atlassian's mission is to unleash the potential of every team.
- We have developed well-known collaborative products, such as Jira, Confluence, and Bitbucket.
- Atlassian helps teams everywhere change the world, powering revolutions across many industries.
- Our organization is dedicated to driving AI innovation across all Atlassian products and platforms.
- 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.
- Our tools are designed to integrate seamlessly with other Atlassian platforms, including the Atlassian Data Platform.
- We believe that the unique contributions of all Atlassians create our success.
Equal Opportunity
- 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.
- Identity verification (which may include use of biometric data) is a condition of employment with Atlassian for employment fraud purposes.
