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

Atlassian; Mountain View (CA); Remote; USD 130,000 - 190,000; Full time

Job summary

An innovative firm is seeking a Senior Principal Machine Learning Engineer to lead the optimization of large language models. This role involves spearheading advanced data analysis, collaborating with cross-functional teams, and driving AI innovation within the organization. You'll have the opportunity to work with cutting-edge technologies and mentor fellow engineers, all while making a significant impact on how teams collaborate globally. If you are passionate about machine learning and eager to shape the future of intelligent solutions, this position offers an exciting challenge in a dynamic environment.

Qualifications

• 8+ years of experience in machine learning with a focus on large-scale model development.
• Deep expertise in LLM and transformer architectures like GPT, BERT, T5.

Responsibilities

• Lead fine-tuning and optimization of large language models for diverse applications.
• Collaborate with cross-functional teams to ensure best-in-class model performance.

Skills

Machine Learning

Large Language Models (LLMs)

Python

Problem-solving

Communication

Education

Ph.D. or Master’s degree in Computer Science

Degree in Machine Learning or Artificial Intelligence

Tools

PyTorch

JAX

TensorFlow

AWS

GCP

Azure

Job description

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

Engineering | Mountain View, United States or Remote | Full-Time

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. We’re seeking a highly skilled and experienced Senior Principal 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 teamwork forward across the globe.

Collaborative Innovation: You will work closely with cross-functional teams, including researchers, engineers, designers, 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.

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

Minimum Qualifications:

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

Preferred Qualifications:

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