Top Senior Machine Learning Engineer Jobs in San Francisco, CA

Own and improve Quora’s second-phase ranking models for feed and digest experiences. Build end-to-end machine learning systems covering data pipelines, candidate extraction, feature engineering, model training, and production integration. Identify new product applications for machine learning, develop efficient algorithms with other ML engineers, and contribute methodologies for large-scale recommendation systems.

Build and operate production machine learning systems for Safety AI, including low-latency APIs, data pipelines, model serving, evaluation, monitoring, and rollout infrastructure. Process large-scale camera and telematics data, optimize cloud and edge-to-cloud execution, track model performance and drift, and partner with applied scientists, firmware engineers, platform teams, and product managers to deliver reliable safety features.

Develop and deploy machine learning models, datasets, algorithms, and training recipes to improve AI safety for multimodal LLMs and agentic systems. Work across LLM security, frontier risks, agentic safety, multi-turn evaluation, content safety, hallucinations, and fairness. Apply post-training methods including SFT and RLHF/RLAIF, conduct safety evaluations and research, and collaborate cross-functionally with engineering, data science, and research teams.

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.

Develop and productize machine learning models for autonomous vehicle perception, including LiDAR and camera perception, sensor fusion, detection, classification, segmentation, and tracking. Build scalable data, training, metrics, and performance workflows, and implement efficient C++ code for the NVIDIA DRIVE AV platform. Collaborate across engineering teams, track advances in machine learning, and optimize models for real-time embedded inference.

Lead ML training performance for large-scale VLM, VLA, foundation-model, and reinforcement-learning systems. Design distributed parallelism, mixed-precision strategies, gradient checkpointing, GPU kernels, profiling workflows, and performance benchmarks. Own RL optimization strategy and roadmap, diagnose bottlenecks across compute and networking layers, partner with research and engineering teams on architecture, and mentor engineers.

Build and operate production machine learning systems for Airbnb Payments, including LLM agents, real-time fraud detection, anomaly detection, personalization, and checkout optimization. Design, train, deploy, monitor, and continuously improve batch and streaming AI applications at global scale. Partner with product, engineering, operations, and data science teams, establish model governance and testing standards, and mentor engineers.

Build and scale end-to-end machine learning infrastructure for computer vision and multimodal AI systems. Responsibilities include developing training and deployment pipelines, managing GPU cloud infrastructure, optimizing low-latency model serving, and establishing monitoring and evaluation frameworks for reliable production performance. The role partners with research, computer vision, and product teams to operationalize models across warehouse environments.

Lead design and productionalize ML algorithms and pipelines to detect account-sharing fraud in real time. Build scalable preprocessing, training, deployment, monitoring, and A/B testing solutions with cross-functional teams to optimize detection while minimizing user disruption.

Designs, develops, tests, operates, and maintains production software with full-stack ownership. Builds scalable, reliable, high-performing applications and complex APIs, with emphasis on backend and machine learning systems, GenAI, ML platforms, and embeddings. Conducts code reviews, mentors engineers, analyzes failures, improves reliability, and collaborates with Product, Design, and partner engineering teams using agile and DevOps practices.

Lead end-to-end LLM inference performance across serving systems, models, GPU hardware, and workloads. Define rigorous benchmarks, diagnose bottlenecks from scheduling through kernels and interconnects, optimize single- and multi-node deployments, and deliver production-ready runtimes and configurations. Collaborate with product and infrastructure teams, evaluate inference technologies, and implement runtime fixes. The role requires strong Python engineering, production experience with serving engines such as vLLM or SGLang, GPU profiling, and knowledge of modern inference optimization techniques.

Own the machine learning and applied AI layer of a software platform, from research through production. Build text-processing pipelines, extract structured data from technical documents, develop LLM-powered decision workflows, create evaluation systems, and build NLP classifiers. Lead exploratory research, translate findings into product features, and collaborate with a small engineering team. The role is remote in the US or Canada, with quarterly travel.

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.

Design, train, and validate machine learning models that improve simulated LiDAR, radar, and camera realism for autonomous vehicle simulation. Build datasets from real and simulated sensor data, integrate 3DGS and NeRF reconstruction, develop generative augmentation methods, define sensor realism metrics, reduce sim-to-real gaps, and mentor engineers and interns.

Own and evolve real-time machine learning detection systems for account compromise. Design, train, deploy, and monitor models in production at scale; architect systems for rapid growth; work with customers to understand attacks; and help establish engineering standards on a small, high-autonomy team.

Lead exploration and application of large language models and generative AI for enterprise contact-center automation. Translate research into high-performing, deployable ML systems; contribute across experimentation, implementation, and deployment; collaborate with engineering and product teams; optimize performance; and help shape team strategy, innovation, and professional growth.

Lead exploration and application of large language models and generative AI, translate research into deployable ML systems, and guide strategic direction. Develop solutions across experimentation, implementation, and deployment while collaborating with engineering and product teams to optimize performance and customer outcomes in contact center automation.

Own and improve 2D, 3D, and fusion-based image segmentation models for clinical ultrasound and CT analysis. Build training, evaluation, dataset, and labeling pipelines; collaborate with clinicians, contractors, and data teams; and productionize auditable, low-latency inference services with confidence scoring, drift monitoring, and safe fallbacks. Work under medical-device design controls and support robust modeling across diverse patient anatomy and imaging conditions.

Own the full lifecycle of production classification and detection models, including transformer fine-tuning, entity detection, behavioral risk scoring, training, serving, versioning, evaluation, drift detection, and retraining. Build SageMaker-based ML systems and define output contracts supporting audit and policy enforcement. Lead the evaluation harness and mentor a co-op while operating as the company’s sole ML engineer.

Build and train transformer models for plant intelligence, owning data curation, architecture, training strategies, evaluation, and production deployment. Develop robust model lifecycle systems and training and evaluation infrastructure for rapid iteration. Drive machine learning projects end to end, from data collection through production, while working independently on a small team. Apply strong software engineering and scientific skills to maintain practical ML systems.

Build and adapt AI models for customer-specific scientific workflows. Lead post-training using SFT and reinforcement learning, design evaluation systems, analyze model failures, and drive iterative quality improvements. Partner with AI researchers and software engineers to integrate reliable model capabilities into production workflows. The role requires strong Python and PyTorch skills, deep familiarity with language, multimodal, or agentic models, and the ability to translate customer needs into technical solutions.

Build and ship production AI systems for enterprise portfolio and planning workflows. Develop anomaly and signal detection, LLM-powered insight synthesis, natural-language planning rule translation, evaluation frameworks, and MCP tools. Partner with external AI providers initially, then establish internal standards for prompt design, model selection, agent architecture, reliability, and evaluation. Collaborate closely with product, design, and domain engineering teams.

Own the production lifecycle for machine learning and LLM-backed systems, including training and post-training pipelines, inference serving, model releases, monitoring, retrieval infrastructure, and cost and latency optimization. Partner with AI researchers and engineering teams to move prototypes into reliable production capabilities across customer tenants. Develop ETL and GraphQL features, enforce tenant isolation, and support deployments in regulated or customer-controlled environments.

Design, build, deploy, and optimize scalable AI and machine learning systems using large datasets. Develop predictive models, recommendation systems, personalization engines, forecasting solutions, generative AI applications, and production ML pipelines. Collaborate with data, backend, and product teams to integrate models into applications, while monitoring performance, data quality, drift, reliability, and cost. Establish best practices for evaluation, experimentation, reproducibility, and deployment, and research emerging AI technologies.

Own the applied AI layer of a software platform by building production AI services, retrieval systems, structured extraction pipelines, evaluation suites, guardrails, and agentic workflows. Manage model performance, reliability, latency, cost, and monitoring. Translate product goals and user feedback into measurable experiments and shipped improvements while leading complex projects and maintaining dependable live systems.