Top Senior Machine Learning Jobs in San Francisco Bay Area, CA

Lead the design and delivery of scalable machine learning systems and customer-facing generative AI applications for Rovo Chat. Build agent harnesses that improve quality, reliability, and latency; advance state-of-the-art AI capabilities; lead engineers through technical design and launch; and collaborate cross-functionally with teams and internal customers. The role also involves applying a deep understanding of the machine learning project lifecycle and supporting platform-based product development.

Design, develop, train, optimize, and deploy machine learning models, including NER, personalization, predictive, and LLM fine-tuning systems. Collaborate with engineering, product, and analytics teams to integrate AI features into Atlassian products. Conduct experimentation, evaluate model performance, monitor product metrics, and improve scalability and reliability. Contribute to foundational model development and advanced AI initiatives while communicating findings to senior leaders.

Build and lead production machine learning and evaluation systems for Atlassian’s Rovo AI agents. Responsibilities include developing datasets, simulations, automated judges, experimentation platforms, and quality signals; advancing reasoning, planning, retrieval, memory, and tool use; identifying and correcting systemic failures; and balancing quality, safety, latency, reliability, and cost. The role requires hands-on technical leadership, cross-functional influence, architectural decision-making, mentorship, and delivery of measurable customer outcomes at scale.

Build AI agent orchestration, routing, planning, memory, and action systems. Develop and deploy machine learning models for search ranking and proactive suggestions, integrate modern LLMs, and design evaluations to improve quality, speed, and cost. The role requires independent ownership, strong product sense, and collaboration in a fast-paced environment.

Develop and deploy scalable machine-learning solutions, particularly deep-learning forecasting models for company top-line metrics. Lead the full ML lifecycle from problem definition and experimentation through production deployment and improvement. Design production-grade architectures, evaluate and improve models, collaborate with cross-functional teams, explain technical concepts to varied audiences, and mentor junior engineers.

Develop and deploy machine learning and LLM capabilities for agentic search, query understanding, search planning, and retrieval. Build evaluation datasets, improve model behavior through experimentation and fine-tuning, and optimize quality, latency, cost, and reliability. Collaborate across search, product, infrastructure, and AI teams while owning projects through production and mentoring engineers.

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.

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.

Build, deploy, scale, and operate production machine learning models and platforms. Responsibilities include developing ML components, distributed data pipelines, cloud and Kubernetes infrastructure, model monitoring and retraining, CI/CD automation, responsible AI governance, and collaboration with product, data science, and engineering teams.

Develop and improve reliable LLM and deep learning capabilities, build scalable real-time machine learning models, advance Responsible AI practices, and define technical strategies for deep learning, RAG, and LLMs. Collaborate with AI scientists and engineers, conduct ML and reinforcement learning research, fine-tune models, and drive adoption of AI best practices across service teams.

Design, build, deploy, scale, and maintain machine learning models and platforms in production. Responsibilities include developing ML infrastructure, data pipelines, cloud architectures, distributed systems, model monitoring, retraining, testing, CI/CD, and responsible AI practices. The role collaborates with Product, Data Science, and Agile engineering teams while applying advanced modeling techniques and managing large-scale containerized ML systems.

Lead the architecture and development of large-scale machine learning systems, distributed training infrastructure, ranking systems, and production ML services. Drive complex initiatives from concept through deployment, establish engineering standards, partner with product and engineering leadership, and mentor data scientists and engineers. Build reliable, low-latency, high-availability AI platforms using advanced ML frameworks, accelerators, and big data technologies.

Leads Adyen’s Credit Risk Modeling team, overseeing the design, productionization, deployment, monitoring, and improvement of machine learning models and rule-based decision systems for credit underwriting. The manager provides hands-on technical leadership, mentors machine learning scientists and data engineers, shapes architecture and product roadmaps, collaborates cross-functionally, recruits team members, and builds an inclusive, high-performing engineering culture.

Lead automated recommendation research and development through agentic AI, building LLM agents that generate hypotheses, implement changes, run experiments, and learn from results. Develop generative retrieval models using Semantic IDs, take research prototypes into production, improve ML infrastructure integrations, and evaluate system quality and impact. Provide hands-on technical leadership through architecture, implementation, reviews, documentation, mentorship, and cross-functional collaboration.

Build, deploy, and maintain machine learning models powering Snapchat products at scale. Own the full ML lifecycle from data analysis through production deployment, apply modern ML techniques, and collaborate with cross-functional teams to launch ML-driven features. Use AI tools to develop scalable services while ensuring code correctness, security, performance, and architectural quality. Mentor collaborators and solve ambiguous, large-scale problems.

Build scalable systems and machine learning models that transform satellite imagery into labeled geospatial data and production computer vision products. Responsibilities include developing data and annotation pipelines, tracking lineage and versioning, deploying inference services, evaluating models, supporting cloud infrastructure, and improving labeling workflows. The role requires strong Python, cloud, containerization, CI/CD, infrastructure-as-code, production support, and deep learning expertise, with preferred experience in GCP, Vertex AI, spatial databases, and satellite imagery.

Design and build reinforcement learning environments, evaluation systems, reward functions, datasets, data pipelines, and ML infrastructure for computer-aided design tasks. The role converts expert mechanical and electrical engineering data into reliable model-training assets, automates scoring and content workflows, and runs experiments guiding technical and product decisions. Candidates need machine learning experience, physical-engineering domain exposure, engineering design software or vision-based ML familiarity, and hands-on reinforcement learning environment experience.

Own numerical analysis and model parity for compressed, quantized, and optimized autonomous-vehicle models. Develop Jacobian/Hessian sensitivity tools, connect numerical differences to driving behavior, diagnose training instabilities, and build low-overhead observability for distributed training. Investigate floating-point errors, edge cases, and implementation divergence while translating analytical results into deployment decisions and production-ready tooling.

Build and lead large-scale, real-time machine learning systems for Pinterest’s growth content and notification ranking. Drive model improvements through A/B testing, own training and serving infrastructure, establish evaluation and monitoring practices, and guide technical direction across teams. Mentor engineers, conduct rigorous reviews, collaborate cross-functionally, apply state-of-the-art research, and improve engineering quality, system reliability, latency, and cost efficiency.

Leads machine learning engineering teams while designing, deploying, scaling, monitoring, and governing production ML models and platforms. Responsibilities include building distributed ML systems, cloud architectures, data pipelines, CI/CD automation, model retraining, responsible AI practices, and resilient production services. The role combines hands-on technical leadership with mentoring, recruiting, talent development, and cross-functional collaboration with Product and Data Science teams.

Build and operate a scalable LLM inference platform integrated with the MAX stack. Develop distributed, multi-node model deployments, high-throughput batching, KV-cache management, autoscaling, observability, cold-start optimizations, and multi-cloud capabilities. Create Helm charts and Kubernetes operators while collaborating with kernel and generative AI teams to optimize performance. The role requires backend engineering, Kubernetes operations, machine learning experience, and distributed systems expertise.

Design, implement, benchmark, and optimize CUDA-based GPU kernels and custom operators for on-vehicle ML inference. Build tooling for profiling and debugging, integrate kernels into the ML runtime, collaborate with compilers and architecture teams, and deliver high-performance, production-grade libraries meeting strict latency, throughput, and reliability targets.

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.

Build and operate scalable machine learning systems for Atlassian’s Growth organization, including personalized recommendations, ranking, decisioning, experimentation, and customer journey optimization. Partner with product, engineering, data science, analytics, marketing, and sales teams to translate growth opportunities into measurable production capabilities. Develop datasets, features, models, evaluation frameworks, and monitoring systems while optimizing for customer value, fairness, quality, and long-term business outcomes.

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.