Top Senior Machine Learning Engineer Jobs in San Francisco, CA

Develop optimized GPU algorithms and high-performance machine learning solutions for vector search, databases, and data processing. Responsibilities include designing, implementing, benchmarking, debugging, and optimizing C++, CUDA, and Python libraries; developing clustering and visualization algorithms; collaborating cross-functionally; and contributing to open-source projects such as cuML, cuVS, and RAFT.

Design, implement, and deploy production-scale machine learning models to evaluate disputes, chargebacks, and fraud likelihood. Build and manage orchestration layers for multiple models, collaborate with stakeholders to integrate partner/customer data, and improve automated and supervised decision-making for a product-focused ML environment.

Develop software for autonomous vehicles that enables safe, smooth, and predictable driving behavior. Frame complex real-world problems, build algorithms and evaluation tools, scale them across large data pipelines, and optimize them for real-time vehicle operation. Collaborate with multidisciplinary teams on motion planning, robotics, machine learning, and autonomous driving systems.

Design, develop, and deploy machine learning models and pipelines to optimize operations. Lead full ML project lifecycle, ensure model evaluation/monitoring, collaborate cross-functionally, mentor junior engineers, and drive continuous improvement in ML applications and processes.

Develop mathematical models for music preferences, similarity, engagement, personalization, and discovery. Build and deploy scalable real-time recommendation and ranking systems, translate product objectives into metrics, run large-scale experiments and causal analyses, and collaborate with product and research leaders on personalization strategy.

Set technical vision for Pinterest’s Product ML Infrastructure, leading distributed model training, fine-tuning, evaluation, and high-scale CPU/GPU inference. Improve GPU efficiency, data loading, execution, kernels, memory, compilation, quantization, scheduling, and capacity. Build reliable, observable training and serving platforms, drive architecture decisions and migrations, partner with AI/ML teams, mentor senior engineers, and establish operational standards across the organization.

Design and deploy machine learning models for financial document processing, optimize large language models, collaborate with cross-functional teams, and guide junior peers in a fast-paced environment.

Lead the architecture and implementation of a Kubernetes-native AI Factory platform for distributed training, simulation, evaluation, inference, and deployment. Build GPU infrastructure, scheduling, storage, networking, observability, model lifecycle capabilities, and self-service developer workflows across cloud, on-premises, sovereign, and air-gapped environments. Partner with ML researchers and autonomy teams while providing technical leadership on scalable, maintainable AI infrastructure.

Design, build, and operate production agentic AI systems for enterprise clients. Develop autonomous agents using Glean, OpenAI/GPT Enterprise, and AWS or GCP. Own agent workflows from architecture and implementation through deployment, monitoring, evaluation, security, and iteration. Apply tool calling, RAG, multi-step orchestration, memory, prompt engineering, structured outputs, permission-aware retrieval, and cost and latency optimization. Measure reliability and quality using evaluations and maintain enterprise-scale production software.

Designs, evaluates, and improves ranking and recommendation systems through experimentation, feature engineering, and production machine learning. Builds reliable data and feature pipelines, incorporates multimodal signals, manages uncertainty, calibration, and model drift, and monitors system performance. Partners cross-functionally with Engineering, Product, Business, and Analytics to deliver scalable, latency-sensitive solutions with measurable business impact.

Develop and deploy machine learning models and production software for multi-object tracking in autonomous trucks. Responsibilities include sensor fusion across camera, LiDAR, and radar; scalable PyTorch training workflows; tracking algorithm development; large-scale data analysis; C++ and Python engineering; safety and validation strategy; architecture decisions; and cross-functional collaboration. The role also provides technical leadership through design reviews, code reviews, mentoring, and engineering best practices.

Build and scale machine learning systems for frontier AI data generation and evaluation. Responsibilities include efficient agentic evaluations, model routing, fine-tuning and serving smaller open-weight models, predictive task-difficulty modeling, LLM-as-judge measurement, and productionizing research prototypes. The role requires rigorous experimentation, statistical analysis, strong Python and software engineering, and end-to-end ownership of production ML or software systems.

Build and operate a reliable ML platform for large-scale distributed GPU training. Responsibilities include Kubernetes infrastructure, developer-facing CLIs and SDKs, AWS infrastructure, data and artifact management, experiment tracking, model registries, CI/CD, observability, documentation, security guardrails, and platform adoption. The role partners closely with researchers and engineers to improve training speed, reliability, usability, and cost efficiency for autonomous transportation systems.

The Sr. Machine Learning Engineer will develop and deploy ML solutions for healthcare, manage data pipelines, and work with large datasets to enhance healthcare delivery.

Develop and optimize large-scale advertising ranking models for CTR and CVR prediction, calibration, user representations, and sequence modeling. Own machine learning systems end to end, including data pipelines, feature engineering, training, evaluation, deployment, and maintenance. Collaborate with platform, product, data science, and engineering teams to run experiments and improve advertiser outcomes, revenue, and user relevance. Apply deep learning and recommendation techniques within production latency, reliability, and cost constraints.

Design, train, deploy, monitor, and optimize speech and NLP models for a voice-agent platform. Build MLOps pipelines, APIs, and inference services while developing Android and iOS applications that integrate AI, voice, real-time communication, and backend services. Create frontend prototypes and dashboards, conduct model experiments, evaluate vendors, optimize performance, document systems, and maintain engineering quality across collaborative teams.

Design and develop machine-learning approaches for vehicle control, including learned vehicle dynamics models, adaptive controllers, data pipelines, evaluation metrics, and simulation tooling. Integrate learned models into safety-critical closed-loop systems and validate algorithms through offline experimentation, simulation, and on-vehicle testing. Collaborate with engineers and research scientists to combine classical control with AI for scalable self-driving technologies.

Design and build scalable ML infrastructure and backend systems for high-impact marketplace applications. Lead architecture, solve low-latency and high-correctness problems, develop and adopt engineering frameworks, and create production-ready technical roadmaps. Partner with Product, Science, and Operations while mentoring engineers and improving engineering standards. The role requires experience with ML models, foundational models or LLMs, MLOps, Kubernetes, data pipelines, distributed workflows, and large-scale scheduling.

Defines the multi-year technical vision and architecture for foundation models supporting search, recommendations, personalization, and conversational AI. Leads high-impact ML platform initiatives across Mobility and Delivery, drives retrieval, ranking, and LLM system design, and makes build-versus-partner investment decisions. Provides principal-level technical leadership, mentors senior engineers, aligns executives across organizations, and connects model improvements to measurable business outcomes at global scale.

Lead machine learning engineering for fraud prevention and account integrity. Shape technical roadmaps, formulate problems, design and productionize large-scale ML systems, oversee model pipelines and system architecture, conduct A/B testing, and manage rollouts. Mentor and develop MLE talent while applying machine learning, statistics, and optimization techniques to real-world risk and trust challenges.

Build and scale machine learning and backend distributed systems for Uber AI Solutions. Own an AI product vertical end to end, driving technical direction, customer value, revenue, adoption, and market growth. Partner with product and business leaders, guide strategic customer relationships, make decisions in ambiguous environments, and build high-performing teams. The role focuses on production ML, GenAI and LLM systems, evaluation, multimodal data, and other applied AI domains.

Build and operate enterprise AI integration products, including MCP servers, agent-facing APIs, React and TypeScript experiences, semantic retrieval systems, and automated AI quality evaluations. Improve latency, reliability, token efficiency, observability, and security across distributed systems. Lead cross-functional engineering initiatives, mentor engineers, and implement enterprise safeguards such as tenant isolation, audit logging, OAuth, and prompt-injection defenses.

As a Senior AI/ML Platform Engineer, you will architect and scale the ML platform, ensuring efficient ML workflows and collaborating with various teams to drive AI adoption.

Design and validate threat models for agentic AI systems, create adversarial experiments and synthetic environments, develop static and dynamic risk analysis, fine-tune and evaluate models, and build production-ready guardrails, monitoring, dashboards, and evaluation infrastructure. Optimize security systems for latency, cost, reliability, and measurable real-world performance while translating research into product capabilities.

Design, build, and deploy production-grade ML models and scalable pipelines (training, inference, monitoring) using NLP/LLM, classification, recommendation, and prediction techniques. Collaborate with product and engineering, optimize latency and costs, and implement retraining and observability for healthcare workflow automation.