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

n 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/perksandbenefitsto 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.