Principal Machine Learning System Engineer

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. Interviews and onboarding are conducted virtually, a part of being a distributed-first company. **Responsibilities**: **About Atlassian**: Atlassian's mission, "to unleash the potential of every team," serves as the foundation of everything we do. We have developed well-known collaborative products, such as Jira, Confluence, and Bitbucket, that fit into the fabric of teamwork across different types of teams and the processes to help every team succeed. Atlassian helps teams everywhere change the world. Our products power revolutions across many industries, enabling teams to collaborate effectively and achieve their highest potential. Take, for example, NASA launching the Rover on Mars or Cochlear gifting those born deaf with the ability to hear. Your work directly impacts the products they use to promote humanity. **About 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 the integration of Atlassian products with AI solutions beyond the Atlassian ecosystem. **About 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 and reliable tools that empower Atlassian teams to harness the power of AI. These tools will facilitate the development, deployment, measurement, and operation of AI & ML experiences. Our tools are designed to integrate seamlessly with other Atlassian platforms, including the Atlassian Data Platform. This integration enables teams to efficiently and swiftly incorporate AI and ML capabilities into their workflows while strictly adhering to all security and data usage policies. Our primary goal is to deliver a smooth and hassle-free experience for Atlassian users, empowering them to harness the potential of AI and ML without any complications. **About This Role**: As a **Principal Machine Learning System 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, such as Jira and Confluence, to solve their specific challenges in building ML solutions. This may involve curating high-quality ML datasets, fine-tuning open-sourced Large Language Models (LLMs), or accessing proprietary LLMs. Your expertise in both ML and software development expertise will be instrumental in overcoming challenging problems and navigating complex infrastructure and architectural issues. This position offers you the chance to lead projects from the technical design phase all the way to launch. You will partner with various teams and internal stakeholders to achieve impactful results. **In this role, you'll get the chance to**: - Collaborate with your teammates to solve complex problems, from technical design to launch. - Deliver cutting-edge solutions that are used by other Atlassian teams and products to build AI features that reach 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. **On your first day, we’ll expect you to have**: - Extensive experience in building Machine Learning and AI infra/platform/system (generally 5+ years) - ** 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. - ** Proficiency with frameworks and languages**: 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. - ** MLOps and automation**: Deep experience implementing MLOps, CI/CD pipelines, and automation for continuous training, deployment, and monitoring of ML models. **It would be great, but not required if you