PhD AI Research Intern - LLMs & Generative AI

Atlassian; San Francisco (CA); Remote; USD 81,000 - 105,000; Full time

Job summary

At Atlassian, the PhD Research Intern will join the Machine Learning Research team to tackle advanced AI challenges, including LLMs, generative AI, and AI optimization. You will work on training, evaluating, and optimizing models used in enterprise software, collaborating with researchers and engineers across disciplines.

The program is a 12-week paid internship, preference for co-location with Seattle or PST timezone; interviews are virtual and Atlassian hires where they have a legal entity.

Qualifications

• Completed Bachelors degree in Computer Science or a related field.
• Currently pursuing a PhD in Computer Science or a related field at any stage of your doctoral studies.
• Strong foundation in AI/ML, LLMs, modeling and/or optimization techniques.

Responsibilities

• Collaborate cross-functionally with Research Scientists and Machine Learning Engineers to design, implement, and evaluate experiments that advance the performance, efficiency, and scalability of modern ML and LLM systems for our AI products.
• Curate, preprocess, and manage large-scale datasets for training and evaluation, ensuring data quality, diversity, and reproducibility across experiments.
• Conduct continued training, fine-tuning, and alignment of large language models for specialized applications such as conversational AI, summarization, generative search, and multimodal agents.
• Evaluate cutting-edge ML algorithms through rigorous experimentation and provide detailed analyses highlighting performance insights, failure modes, and opportunities for improvement.
• Contribute to publications and presentations at internal workshops or top-tier academic venues, helping to drive innovation in Enterprise AI and large-scale ML systems.

Skills

AI/ML basics

LLMs

Python

NumPy

Pandas

PyTorch

NLP

CV

Publications

Algorithms

Communication

Education

Bachelor's degree in Computer Science or related field

PhD in Computer Science or related field

Tools

Hugging Face

Matplotlib

PyTorch

Job description

At Atlassian, the PhD Research Intern will join the Machine Learning Research team to tackle advanced AI challenges, including LLMs, generative AI, and AI optimization. You will work on training, evaluating, and optimizing models used in enterprise software, collaborating with researchers and engineers across disciplines.

The program is a 12-week paid internship, preference for co-location with Seattle or PST timezone; interviews are virtual and Atlassian hires where they have a legal entity.