Principal Machine Learning Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Atlassian · 23 hours ago

Principal Machine Learning Engineer

Atlassian is on a mission to unleash the potential of every team, investing deeply in Generative AI. They are seeking a Principal Machine Learning Engineer to lead technical innovations in advanced GenAI modeling, applied research, and rapid prototyping, contributing to high-impact solutions across the company.

CollaborationEnterprise SoftwareSaaSSoftware
check
Comp. & Benefits
check
H1B Sponsorednote

Responsibilities

Design and build novel approaches for knowledge-grounded LLMs, embeddings, and multimodal modeling
Explore fine-tuning, post-training, and retrieval-augmented strategies to adapt foundation models to Atlassian’s data
Lead technical explorations that expand the frontier of what’s possible in applied generative modeling
Drive POC experimentation, from agentic workflows to novel GenAI-powered assistants
Translate research ideas into fast, working prototypes that demonstrate user value
Define and implement evaluation methods (groundedness, hallucination detection, human-in-the-loop) to measure success
Provide technical mentorship to team members
Influence design and implementation across retrievers, rankers, orchestration, and serving infrastructure
Collaborate with product and platform teams to turn prototypes into scalable systems
Contribute to the roadmap for advanced GenAI modeling and applied research
Partner with peers across Rovo, Jira Search, Knowledge AI, and Platform to drive cross-team impact
Champion responsible AI, groundedness, and data quality in all innovations

Qualification

Machine Learning EngineeringLLMsNLPGenAI ecosystemsPrototypingRetrieval-augmented generationFine-tuning frameworksScaling ML systemsTechnical leadershipMentoringCollaboration

Required

6+ years in ML/AI engineering, with deep expertise in LLMs, NLP, or search
Proven track record of building and shipping ML-powered or GenAI-integrated products
Hands-on experience with retrieval-augmented generation (RAG), embeddings, hybrid retrieval, or agentic workflows
Strong background in scaling ML systems from prototype to production
Expert-level knowledge of GenAI ecosystems (OpenAI, Anthropic, Mistral, OSS) and fine-tuning/serving frameworks
Deep technical skills across modeling, infrastructure, and orchestration (LangChain, LlamaIndex, vector DBs)
Strong prototyping ability, balancing speed with rigor in experimentation
Ability to set technical direction and inspire peers with thought leadership
Master's or PhD in Computer Science, Machine Learning, or related fields

Preferred

Experience with multimodal models, knowledge graphs, or semantic embeddings
Familiarity with evaluation metrics for search/GenAI (NDCG, groundedness, human preference modeling)
Contributions to open-source ML frameworks or GenAI research communities
Passion for mentoring and fostering innovation

Benefits

Health and wellbeing resources
Paid volunteer days
And so much more

Company

Atlassian

company-logo
Atlassian is a software company that offers proprietary software products for teamwork, project management, and software development.

H1B Sponsorship

Atlassian has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (351)
2024 (184)
2023 (190)
2022 (259)
2021 (156)
2020 (162)

Funding

Current Stage
Public Company
Total Funding
$210M
Key Investors
T. Rowe PriceAccel
2015-12-10IPO
2014-04-08Secondary Market· $150M
2010-07-14Series A· $60M

Leadership Team

leader-logo
Michael Cannon-Brookes
Co-founder & Co-CEO
linkedin
leader-logo
Scott Farquhar
Co-Founder and Board Member
linkedin
Company data provided by crunchbase