Senior Machine Learning Engineer

About the Position

Atlassian is seeking a Senior Machine Learning Engineer to join our Agentic Search team. You’ll build agentic search systems that help people and AI agents discover relevant enterprise knowledge, reason through complex questions, and accomplish tasks across products and tools.

Responsibilities

• Build and deliver agentic search capabilities for AI assistants and developer workflows through APIs, command-line tools, and protocols such as the Model Context Protocol (MCP).
• Develop reproducible benchmarks and evaluation tools that reflect realistic tasks across models and agent environments.
• Use evaluation results, production signals, and developer feedback to improve the full search system.
• Build agentic search systems that interpret user intent, plan and execute searches, select appropriate sources and tools, and adapt based on retrieved evidence.
• Explore and apply advances in LLMs and agent systems, including prompting, model selection, training data improvements, and fine-tuning where appropriate.
• Own projects from problem definition and technical design through experimentation, deployment, and ongoing measurement.
• Collaborate across product, search, and AI teams to shape technical direction and integrate agentic search into customer experiences.
• Contribute to technical design and code reviews, mentor engineers, and help the team evolve its engineering practices.

Requirements

• A bachelor’s or master’s degree in Computer Science or a related field, or equivalent practical experience.
• 4+ years of relevant industry experience in machine learning, including delivering ML capabilities into production.
• Strong Python programming skills and the ability to build reliable, maintainable production systems.
• Experience in one or more of LLM applications, AI agents, information retrieval, search relevance, or natural language processing.
• Experience designing experiments, building evaluation datasets, and analyzing model and system behavior to guide improvements.
• Regular use of AI tools in your daily engineering workflow, such as coding, prototyping, debugging, or experimentation.
• An understanding of the ML development lifecycle, from data preparation and modeling to deployment, monitoring, and iteration.
• The ability to lead ambiguous technical projects, learn new approaches quickly, make practical tradeoffs, and communicate clearly with engineering and product partners.

Nice to Have

• Experience building agents that use tools, multi-step search systems, or retrieval-augmented generation applications.
• Experience designing APIs, developer tools, or interfaces that make search and knowledge accessible to AI agents, including MCP.
• Experience developing agent evaluations or reproducible benchmarks, including trajectory analysis, human evaluation, or model-based judging.
• Experience with semantic or hybrid retrieval, ranking, or context-aware search.
• Experience with LLM fine-tuning or post-training, including supervised fine-tuning, preference optimization, or reinforcement learning.
• Experience with distributed data processing and cloud ML environments such as Spark, AWS, or Databricks.

We Offer

• Competitive salary ranging from $171,063 - $269,075 per year, depending on the geographic pay zone.
• A wide range of perks and benefits designed to support you, your family, and to help you engage with your local community.
• Flexible work options, including remote work opportunities.

About the Company

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.