Senior Principal Machine Learning Engineer - LLM Post-Training and Optimization

Atlassian is a global leader in team collaboration and productivity software, empowering teams to unleash their potential and achieve remarkable results. Our suite of tools, including Jira, Confluence, Trello, Bitbucket, and Rovo, is trusted by millions of users worldwide to plan, track, and manage their work effectively. At Atlassian, we are dedicated to building innovative tools and technologies that drive the future of work. We're seeking a highly skilled and experienced Senior Principle Machine Learning Engineer to propel our efforts in Large Language Model (LLM) post-training and optimization, shaping the future of intelligent, team-centric solutions.
Your Role and Impact:

• LLM Post-Training and Optimization: Spearhead the optimization and enhancement of cutting-edge large language models (LLMs) to deliver superior performance and efficiency. Your work will span fine-tuning, optimization, and deployment of advanced LLMs tailored to a variety of applications.
• Advanced Data Analysis: Dive deep into vast datasets to extract meaningful insights. Your work will power the AI models to understand complex user interactions and content within Atlassian tools to propel team work forward across the globe.
• Collaborative Innovation: You will work closely with cross-functional teams, including researchers, engineers, designers, engineering and product managers, to ensure our models deliver best-in-class performance, efficiency, and scalability.
• AI Evangelism: Be the voice of AI within the team. Educate and inspire your colleagues about the potential of AI, leading workshops and knowledge-sharing sessions.

What you'll do

• Lead the fine-tuning and post-training optimization of large language models (LLMs) for diverse applications.
• Develop and implement techniques for model compression, quantization, pruning, and knowledge distillation to optimize performance and reduce computational costs.
• Conduct research on advanced techniques in transfer learning, reinforcement learning, and prompt engineering for LLMs.
• Design and execute rigorous benchmarking and evaluation frameworks to assess model performance across multiple dimensions.
• Collaborate with infrastructure teams to optimize LLM deployment pipelines, ensuring scalability and efficiency in production environments.
• Stay at the forefront of advancements in LLM technologies, sharing insights, driving innovation within the team, and leading agile development.
• Mentoring other team members, facilitating within/across team workshops, fostering a culture of technical excellence and continuous learning.

We'll expect you to have

• Ph.D. or Master's degree in Computer Science, Machine Learning, Artificial Intelligence, or a related field.
• 8+ years of experience in machine learning, with a focus on large-scale model development and optimization.
• Deep expertise in LLM and transformer architectures (e.g., GPT, BERT, T5).
• Strong proficiency in Python and ML frameworks such as PyTorch, JAX, or TensorFlow.
• Experience with distributed training techniques and large-scale data processing pipelines.
• Proven track record of deploying machine learning models in production environments.
• Familiarity with model optimization techniques, including quantization, pruning, and knowledge distillation.
• Strong problem-solving skills and ability to work in a fast-paced, collaborative environment.
• Excellent communication skills and ability to translate technical concepts for diverse audiences.

It's great, but not required, if you have

• Experience with multi-modal LLMs or domain-specific fine-tuning.
• Knowledge of cloud-based ML platforms (e.g., AWS, GCP, Azure).
• Contributions to open-source ML projects or publications in top-tier conferences.
• Familiarity with MLOps practices and tools.

Compensation
At Atlassian, we strive to design equitable, explainable, and competitive compensation programs. To support this goal, the baseline of our range is higher than that of the typical market range, but in turn we expect to hire most candidates near this baseline. Base pay within the range is ultimately determined by a candidate's skills, expertise, or experience. 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: $292,900 - $407,200
Zone B: $263,700 - $366,500
Zone C: $243,100 - $338,000
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.
Our perks & benefits
Atlassian offers a variety of perks and benefits to support you, your family and to help you engage with your local community. Our offerings include health coverage, paid volunteer days, wellness resources, and so much more. Visit go.atlassian.com/perksandbenefits to learn more.
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 .