Machine Learning Engineer Jobs in San Francisco

NVIDIA is seeking a Senior Verification Engineer to join their Display Team, focusing on verifying the world's foremost GPU and SoC display engines. In this role, you will collaborate with Architects and ASIC Engineers to own testplan development, verification sign-off, and infrastructure. The goal is to deliver high-quality IPs that power NVIDIA products across various markets, including consumer, professional, automotive, and AI. You will leverage agentic AI to enhance efficiency in tasks like testplan development, automation, and debug.

$168k - $311k

Step into a role where you can make a significant impact on the future of autonomous vehicles. As a Senior Software Engineer, ML Loop - Automotive at NVIDIA, you will be at the forefront of our data flywheel operation. This is your opportunity to work with hardworking individuals and pioneer innovative technologies to contribute to world-class solutions. Embrace this outstanding opportunity to be part of NVIDIA's ambitious journey and help craft the future of autonomous driving through outstanding data engineering.

$168k - $270k

NVIDIA is seeking a Deep Learning Compiler Engineer to join its Deep Learning Compiler (DLC) team. This role involves developing and optimizing compiler algorithms for deep learning networks, which are crucial for NVIDIA's inference engine across various platforms. The compiler is designed to deliver leading inference performance, fast build times, reduced memory footprints, and ease of use for the global deep learning community. You will collaborate with deep learning software framework and hardware architecture teams to accelerate the next generation of deep learning software, contributing to defining public APIs, performance analysis, and implementing compiler infrastructure techniques.

$152k - $242k

NVIDIA is seeking an extraordinary Senior Perception Engineer to develop and productize autonomous driving solutions. This role focuses on creating end-to-end perception modules responsible for understanding traffic signals, signs, road markings, and construction objects, crucial for driving without reliance on HD maps. The goal is to enhance the robustness, accuracy, and efficiency of these solutions to enable autonomous driving in all conditions. You will be instrumental in designing perception systems, applying deep learning for various traffic-related tasks, developing generalizable approaches for diverse driving scenarios, and driving data-driven development by collaborating with data collection and labeling teams. Leveraging data simulation and augmentation will be key to solving extreme scenarios, and you will be responsible for productizing these solutions to meet stringent safety, latency, and software robustness requirements.

$184k - $357k

NVIDIA Research is seeking leading researchers in the area of simulation and AI for quantum. You should have a strong background of research excellence, software systems building experience, and interest in advancing the state-of-the-art in quantum computing and applications of GPUs to the quantum computing ecosystem. Specific areas of research interest include but are not limited to bespoke quantum representations, tensor network methods, quantum surrogate modeling, quantum inverse design, and automated proving for quantum systems. This role involves researching and developing creative and innovative algorithms and software systems for the quantum computing ecosystem, collaborating with research and product teams, and publishing original research.

$0k - $0k

NVIDIA Research is seeking PhD-level students for a research internship focused on AI hardware/software co-design, hardware accelerator architecture, IC design methodology, and VLSI design. The ideal candidate will possess a broad understanding of machine learning fundamentals, quantization and numerical methods for model optimization, digital VLSI circuits, and advanced VLSI design and verification methodologies, including the application of agentic AI to hardware design. This role involves contributing to cutting-edge research in AI accelerators and VLSI design, exploring innovative numerical methods, and potentially collaborating on research prototype testchips.

$0k - $0k

NVIDIA is seeking strategic, ambitious, hard-working, and creative individuals for its 2027 Deep Learning Computer Architecture internships. This 12-week full-time program offers an excellent opportunity to expand your career and gain hands-on experience with industry-leading Deep Learning Computer Architecture teams. Interns will work on projects with a measurable impact on NVIDIA's business, tackling complex challenges in AI and digital twins that are transforming major industries.

$0k - $0k

NVIDIA is seeking strategic, ambitious, hard-working, and creative individuals for its 2027 Deep Learning internship program. This 12-week full-time internship offers an excellent opportunity to expand your career and gain hands-on experience with industry-leading Deep Learning teams. Interns will work on projects with a measurable impact on NVIDIA's business, contributing to advancements in AI and digital twins that are transforming major industries. We are looking for students pursuing a B.S., M.S., or Ph.D. degree in a relevant field who are passionate about tackling complex challenges.

$0k - $0k

Join a team that analyzes large-scale datacenter workloads on GPU-accelerated clusters. You will transform telemetry and workload data into clear findings and visuals, partnering with OS, container, GPU, and systems engineers. When useful, you will apply machine learning and deep learning techniques for categorization and forecasting, coordinating these into tools the team uses. This role involves analyzing signals to identify opportunities for application and platform improvements, working with high-dimensional data to spot trends, and communicating conclusions to engineers and leadership.

$152k - $242k

NVIDIA is seeking a Senior Software Engineer to join our CSP Engagements team, focusing on system software for Datacenter products such as GB200. This role involves deep technical expertise in embedded firmware, Linux kernel development, and middleware development, combined with customer-facing responsibilities to enable cloud service providers with next-generation computing platforms. You will work at the intersection of hardware and software, driving technical solutions from concept through deployment.

$184k - $357k

NVIDIA is seeking a Senior Systems Software Engineer to join their Deep Learning Solutions team. This role focuses on optimizing deep learning inference for autonomous vehicles and robotics on edge devices. The ideal candidate will be a hands-on specialist capable of analyzing model architectures at the operator level, identifying performance bottlenecks through kernel trace analysis, and evaluating modern AI architectures on GPU and SOC platforms. This position directly contributes to enhancing real-time perception and response capabilities for autonomous systems, working at the intersection of model frameworks, compiler technology, and embedded hardware. Collaboration with automotive OEMs, robotics teams, and internal hardware groups is a key aspect of this role.

$224k - $357k

We are seeking senior systems software engineers to enhance the performance, efficiency, and readiness of the CUDA Driver for next-generation accelerated computing. This role involves developing performance-critical CUDA Driver features and systems software improvements to maximize the potential of NVIDIA GPUs for AI, deep learning, HPC, and other CUDA-powered applications. You will tackle complex performance challenges at the intersection of systems software, CPUs, interconnects, and GPUs, tracing issues from real workloads through the entire stack, designing and shipping production features, and delivering validated optimizations for current and emerging platforms. This position offers the opportunity to combine hands-on engineering with broad technical influence, collaborating with various CUDA software, hardware architecture, frameworks, applications, product, and customer-facing teams. You will lead cross-layer investigations, mentor engineers, and help define performance direction, shaping both current GPU utilization and future CUDA software and GPU architectures.

$184k - $357k

NVIDIA is seeking a Senior Machine Learning Engineer to join its prediction team for autonomous driving solutions. This role involves developing and productizing cutting-edge prediction modules responsible for generating multi-modal scene-consistent predictions for road participants. The engineer will focus on enhancing the robustness, accuracy, and efficiency of these solutions to enable autonomous driving in diverse environments and conditions. The position offers the opportunity to work on state-of-the-art ML techniques and contribute to the future of autonomous driving.

$224k - $357k

NVIDIA is building GeForce G-Assist, an on-device AI assistant that integrates Small Language Models (SLMs), retrieval systems, and hybrid cloud capabilities for responsive, context-aware assistance within the GeForce ecosystem. This role involves close collaboration across engineering and product teams to ensure G-Assist performs reliably in real-world scenarios. The focus is on production model behavior rather than just benchmarks, with an emphasis on accuracy, robustness, and conversational reliability. The team works with SLM and VLM architectures, hybrid cloud/local model architectures, and optimizes local inference for performance.

$152k - $288k

Stripe

Growth Platform at Stripe builds machine learning systems to help businesses discover and utilize Stripe products. These recommendations are integrated across various platforms including the Dashboard, email, onboarding, documentation, and AI agent interfaces. The team combines business understanding with models to determine the most useful recommendations, optimal timing, and effective learning from outcomes. Their work encompasses recommendation and ranking models, contextual bandits, agent-based recommendations, and the underlying data and evaluation systems, aiming to drive business growth through relevant product adoption while maintaining a positive user experience.

Salary not listed

NVIDIA is seeking a Software Engineer, Performance Instrumentation to join their team in Santa Clara, CA. This role offers an exciting opportunity to contribute to groundbreaking projects at the intersection of AI and accelerated computing. You will be instrumental in defining and maintaining performance data requirements, identifying and addressing visibility gaps, and ensuring the quality and efficiency of instrumentation APIs. This position requires a strong background in software engineering with a focus on performance and data quality within machine learning or systems software environments.

$168k - $270k

Glean is seeking Machine Learning Engineers to join the Enterprise Brain team, focusing on building proactive AI products that revolutionize enterprise workflows by automating tasks and enhancing user productivity. This role involves leveraging deep user understanding and state-of-the-art enterprise graphs, utilizing LLMs, advanced ML techniques, agent orchestration, and cutting-edge ranking methods. You will contribute to a secure, open platform that redefines how enterprises find, use, and act on knowledge, with a focus on enterprise trust and scalability.

$200k - $280k

Moveworks is seeking a Senior Staff ML Engineer to join our AI Safeguards team. In this role, you will be instrumental in designing, building, and shipping critical safeguards within our agent harness. Your work will focus on protecting against prompt injection, single and multi-turn jailbreaks, toxicity, sensitive topics, and sensitive data exposure. You will develop robust evaluation frameworks to measure the efficacy of these safeguards, addressing both false positives and false negatives using diverse datasets. Collaboration with our Security team will be key, translating their findings into actionable eval cases and fixes, and co-developing mitigations. You will also select, integrate, and fine-tune detection models within a shared stack serving both commercial and government environments, while carefully balancing quality, latency, and cost for controls operating in the request path. Additionally, you will build configurable safeguard policies to empower customers in managing sensitive topics and monitor safeguard behavior in production, addressing any regressions.

Salary not listed

MatX is building the compute platform for AGI, developing vertically integrated full-stack solutions from silicon to systems. We focus on hardware and software to train and run the largest ML workloads for AGI, primarily using the Rust programming language. This role involves designing and optimizing kernels that interface directly with our hardware, working in partnership with ML Research and Hardware Engineering teams, and providing expertise on hardware architecture from a programmer's perspective to ensure seamless integration with the software stack.

$160k - $275k

Figure is an AI robotics company developing autonomous general-purpose humanoid robots with the goal of shipping robots with human-level intelligence for home and commercial markets. The Helix team is responsible for developing the core AI systems that power humanoid autonomy. We are seeking a Helix AI Engineer, Video Pretraining to lead the development of large-scale video foundation models trained on diverse real-world and robot-collected data. This role focuses on pretraining models that learn from raw video, capturing motion, interaction, and temporal structure to enable downstream capabilities in perception, prediction, and embodied reasoning.

$200k - $400k

Sift

As a Machine Learning Engineer at Sift, you will bridge the gap between data science and large-scale distributed systems. You will build end-to-end pipelines that extract signals, train custom models per merchant, and serve predictions at production scale with low latency. You will work on an automated machine learning ecosystem that dynamically recalibrates models based on streaming global telemetry data.

$170k - $240k

NVIDIA is seeking a Senior Validation Engineer to join the DGX Server Product Engineering Team. In this role, you will collaborate with a team of hardware and software engineers to develop and implement sophisticated automated test plans for our leading GPU-accelerated computing products. You will be instrumental in ensuring the performance and reliability of cutting-edge AI hardware by developing and tuning workloads, adapting state-of-the-art AI training and inference frameworks, and applying architectural analysis to characterize workload behavior and identify bottlenecks. This position offers the opportunity to make a significant impact on the future of computing by working on AI-driven technologies.

$136k - $259k

Salesforce is seeking a highly motivated, hands-on Lead Machine Learning Engineer to define and execute the technical ML strategy for a foundation machine learning platform team. This role involves full lifecycle development and optimization of ML pipelines, with a strong focus on MLOps, infrastructure-as-code, CI/CD, and thorough monitoring. The lead will manage multiple ML pipelines, work closely with cross-functional teams, mentor others, and requires deep expertise in various ML technologies to deliver measurable business impact within a security and compliance framework. A critical component of this role is the proven ability to productize Generative AI solutions and agentic workflows to enhance the security posture and resilience for the organization and all Salesforce customers.

$173k - $260k

Figure is an AI robotics company developing autonomous general-purpose humanoid robots with the goal of shipping robots with human-level intelligence. These robots are engineered to perform a variety of tasks in both home and commercial markets. We are seeking a Reinforcement Learning Engineer to focus on the development, training, deployment, and evaluation of advanced reinforcement learning algorithms specifically for the whole-body control of our humanoid robots.

$150k - $350k