Top AI & Machine Learning Jobs

Lead and grow a high-performing ML team to design and deliver end-to-end AI features for DevAI products. Drive LLM integration, prompt engineering, technical strategy, project delivery, stakeholder alignment, and process optimization while mentoring engineers and managing performance.

As a Principal Machine Learning Systems Engineer, you will lead the development and implementation of machine learning systems and solutions to enhance team collaboration and productivity across Atlassian's software products.

Design, build, and operate scalable, low-latency search and retrieval systems powering Atlassian products. Develop and serve production ML models (neural rankers, embeddings, rerankers) using PyTorch and Triton. Implement vector indexing and RAG/agentic retrieval pipelines, ensure observability and cost optimization, manage on-call/runbooks, and collaborate cross-functionally while mentoring junior engineers.

Set technical vision and build AI foundations used across Atlassian: prototype and write code, build scalable AI infrastructure, measure AI effectiveness, provide context and cost-efficient compute, partner with product teams, and mentor senior engineers.

Lead and grow an engineering team building enterprise-grade agentic AI integrations. Define technical strategy for multi-agent systems (MCP, A2A), collaborate cross-functionally, maintain highly reliable, low-latency services, and drive industry-spec contributions and interoperability.