Top AI & Machine Learning Jobs

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 engineering teams building large-scale AI/ML platform infrastructure (model serving, training, feature stores). Drive technical strategy, deliver complex backend systems to production, partner with product and stakeholders, uphold high engineering and operational standards, and coach engineers for career growth.

Lead design and deployment of production-grade ML systems (ranking, retrieval, LLMs). Drive architecture decisions, run experiments and evaluations, build scalable feature pipelines and monitoring, mentor engineers, and collaborate cross-functionally to integrate AI into products and scale ML capabilities.

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.