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

Atlassian · 5 hours ago

Principal Machine Learning Engineer

Atlassian is on a mission to unleash the potential of every team, investing in Generative AI to enhance team collaboration. They are seeking a Principal Machine Learning Engineer to lead technical explorations and drive innovations in advanced GenAI modeling and applied research.

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
Act as the technical anchor of a small tiger team (manager + ICs), mentoring engineers and scientists
Influence design and implementation across retrievers, rankers, orchestration, and serving infrastructure
Collaborate with product and platform teams to transition prototypes into scalable systems
Contribute to the roadmap for the Forge as a global hub 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

ML/AI engineeringLLMs expertiseGenAI ecosystemsRetrieval-augmented generationNLP expertiseScaling ML systemsPrototyping abilityThought leadershipMentoringCollaboration

Required

Extensive experience in ML/AI engineering (generally 6+ years), 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 field preferred—or equivalent industry experience

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, fostering innovation, and shaping a new AI hub in Singapore

Benefits

Health and wellbeing resources
Paid volunteer days

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 (263)
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
CEO and Co-Founder
linkedin
leader-logo
Scott Farquhar
Co-Founder and Board Member
linkedin
Company data provided by crunchbase