Top AI & Machine Learning Jobs in San Francisco, CA

Build and operate scalable machine learning systems for personalized recommendations, growth journeys, cross-product expansion, and sales experiences. Design models, datasets, features, evaluation frameworks, and decisioning services; run offline policy evaluations and online experiments; and monitor attribution, latency, quality, and fairness. Partner with product, engineering, data science, analytics, marketing, and sales teams to turn ambiguous growth opportunities into measurable production capabilities and strategic ML initiatives.

Lead the technical direction for Atlassian’s Search Serving platform, building globally scalable, low-latency agentic search systems. Own architecture for retrieval, indexing, query processing, and ML inference while improving quality, reliability, scalability, operability, and cost. Establish production ML practices for evaluation, experimentation, observability, rollouts, and incident response. Optimize inference for embedding, retrieval, and ranking models, and lead cross-team initiatives while mentoring senior engineers and aligning technical leaders.

Build an AI-powered marketing insight workflow using Rovo and Atlassian data. Responsibilities include developing agents to analyze OKRs, detect trends, draft executive summaries, flag low-performing results, generate leadership prompts, and create CMO pre-reads. The intern will establish human review and quality assurance processes, document reusable workflows, explore related automations, and measure time savings, adoption, and decision-making impact.

Lead data science efforts to inform product and business strategy through quantitative analysis, experimentation, forecasting, and storytelling. Partner with cross-functional teams to define hypotheses, measure impact, influence roadmaps, present insights to senior leadership, and monitor key product metrics to drive value.

Build and maintain core infrastructure for developing, training, evaluating, deploying, and operating machine learning models and pipelines for Atlassian’s GenAI products. Solve complex software infrastructure and architecture challenges, lead engineers through technical projects from design to launch, and collaborate with internal teams and customers. The role involves large-scale distributed systems, data processing, cloud platforms, CI/CD, and machine learning lifecycle support.

Lead the design and delivery of knowledge graph and machine learning systems that infer personal work context from connected tools. Build graph inference pipelines, schemas, permission-aware APIs, Rovo Chat integrations, and MCP-compatible CLI experiences. Establish evaluation frameworks, improve retrieval and grounding, provide technical direction across teams, mentor engineers, and ensure responsible, privacy-safe AI systems operate reliably at production scale.

Develop and productionize machine learning systems for search, retrieval, ranking, recommendations, conversational experiences, and AI agents. Own projects across the ML lifecycle, build evaluation datasets and benchmarks, conduct offline and online evaluations, analyze model quality, and improve system performance. Design scalable architectures addressing reliability, latency, privacy, and cost. Collaborate across product and engineering teams, communicate technical decisions, mentor engineers, and contribute to ML best practices.

The job description does not include specific responsibilities or qualifications. The role is identified as Principal Data Scientist and may be performed remotely, in an office, or in a hybrid arrangement, depending on the employee’s location and Atlassian’s legal entity availability.

Lead the design, development, deployment, and optimization of production machine learning systems for search, recommendations, personalization, NLP, deep learning, and AI applications. Own projects across the full ML lifecycle, design scalable architectures and evaluation strategies, collaborate with product and engineering teams, translate research into reliable systems, and mentor other engineers.