Atlassian

Principal Machine Learning Systems Engineer (GenAI Products & Knowledge Innovations)

Atlassian
Apply
22 hours ago
Remote, United States +3 moreStaff+
H1B sponsor

Base Salary

$196k - $309k/yr

Responsibilities

  • Architect and implement scalable systems for training, fine-tuning, and serving large language models and embeddings.
  • Build retrieval, hybrid search, and RAG pipelines using knowledge-grounded data.
  • Develop infrastructure and tools for rapid experimentation, evaluation, and deployment of prototypes.
  • Partner with applied scientists to convert new ideas and proof-of-concept systems into robust, production-ready pipelines.
  • Optimize latency, throughput, and resource efficiency for generative AI workloads.
  • Collaborate with ML engineers, backend developers, and product teams to ship end-to-end innovations.
  • Contribute to model deployment, monitoring, evaluation, and ML systems reliability practices.

Requirements

  • At least 6 years of experience in ML systems engineering, backend engineering, or infrastructure roles.
  • Strong track record of building and scaling ML-powered services in production.
  • Experience with large-scale model training, inference pipelines, or search and retrieval systems.
  • Proficiency with Python, PyTorch, TensorFlow, and Hugging Face.
  • Experience with Weaviate, Pinecone, or FAISS vector databases and LangChain or LlamaIndex orchestration frameworks.
  • Strong coding skills and ability to optimize systems for performance and reliability.
  • Familiarity with AWS, GCP, Azure, Kubernetes, and Docker.
  • Bachelor’s or master’s degree in computer science, machine learning, or a related field, or equivalent industry experience.
  • Preferred experience in distributed systems, high-performance computing, GPU optimization, GenAI evaluation metrics, monitoring, observability, reliability practices, or open-source infrastructure and ML systems frameworks.

Benefits

  • Flexible choice to work from an office, home, or a combination of the two, subject to availability in countries where Atlassian has a legal entity.
  • Health and wellbeing resources.
  • Paid volunteer days.
  • Potential eligibility for benefits, bonuses, commissions, and equity.
Atlassian

About Atlassian

10,000+ employees

Atlassian powers the collaboration that helps teams accomplish what would otherwise be impossible alone. From space missions and motor racing to bugs in code and IT requests, no task is too large or too small with the right team, the right tools, and the right practices. Over 300,000 global companies and 80% of the Fortune 500 rely on Atlassian’s software, like Jira, Confluence, Loom, and Trello, to help their teams work better together and deliver quality results on time. With our 300,000+ customers and team of 10,000+ Atlassians, we are building the next generation of team collaboration and productivity software. We believe the power of teams has the potential to change the world — one that is more open, authentic, and inclusive.

Contact me