AI Engineering Jobs in San Francisco

The AI revolution is driven by computation that is fast, efficient, and economical. At NVIDIA, performance is paramount, and the CUDA driver is a critical layer where architectural invention becomes useful computing. This team has a unique opportunity to impact numerous downstream systems and products by optimizing fundamental aspects of the CUDA driver and runtime. We are seeking systems software engineers to design and implement production C/C++ features and optimizations within the CUDA driver and runtime. You will trace workloads across various system boundaries, bring up new platforms, and translate evidence into future software and hardware direction, ultimately making AI tools more responsive and efficient.

$124k - $196k

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 Architects, Functional Modeling Engineers, and Simulation experts to join its Architecture Modeling group. In this role, you will collaborate with world-class architects to enable functional simulation platforms across GPU generations. You will be instrumental in modeling GPU architecture, baseboard components, and other features, working within a matrixed environment to document, design, develop tools for analysis and simulation, and validate models. The position involves familiarizing yourself with various functional and performance simulation models across NVIDIA, developing tests and testing infrastructure for new architectures, and guiding the improvement of the simulation platform to support future GPU architectures. You will also have the opportunity to develop or use AI to assist with daily tasks.

$152k - $288k

NVIDIA is seeking a Principal AI and ML Infra Software Engineer to join their Hardware Infrastructure team. This role is pivotal in enhancing researcher efficiency by implementing advancements across the entire technology stack. The engineer will collaborate closely with customers to identify and resolve infrastructure deficiencies, enabling groundbreaking AI and ML research on GPU Clusters. The goal is to build powerful, effective, and scalable solutions that shape the future of AI/ML technology.

$272k - $431k

We are hiring senior engineers to work on the CUDA driver, a core component of our platform for accelerating general purpose computation on the GPU. Our team delivers features and improvements to better realize the potential of NVIDIA hardware for a growing range of computational workloads, ranging from deep learning, scientific computation, and self-driving cars to video games and virtual reality! CUDA defines a unified programming model across a range of system configurations and hardware capabilities. To accomplish this, the CUDA driver interacts with GPU hardware, kernel mode drivers, switches and the operating system. As a member of our team, you will use your design abilities, coding expertise, and creativity to deliver the best Compute platform in the world. You will craft elegant solutions to exciting problems and craft the future direction of CUDA as you collaborate with your peers across NVIDIA.

$272k - $431k

NVIDIA is seeking talented individuals to join our Architecture Modeling group, focusing on GPU and SOC architecture. In this role, you will collaborate with world-class architects to integrate state-of-the-art Generative AI applications into simulation platforms and engineering workflows. You will develop and deploy scalable GenAI applications, leverage AI tools to accelerate modeling, and work with hardware architects to design and deploy AI-based solutions tailored to specific problem domains. Additionally, you will collaborate with infrastructure engineers to enhance automated workflows using LLMs and research emerging AI technologies to maintain a competitive edge.

$152k - $288k

We are seeking a Senior Power Architecture & Optimization Engineer to advance energy efficiency through sophisticated analytics and AI, including specialized LLMs for power analysis. This role is central to NVIDIA's mission of creating the most energy-efficient GPUs and SoCs, leveraging AI and LLMs as a key differentiator for future products. You will combine expertise in power architecture with machine learning, reinforcement learning, data analytics, and large language models to innovate in modeling, analyzing, and optimizing energy consumption across various design levels.

$136k - $265k

NVIDIA is seeking a Principal Engineer to lead critical platform initiatives at the intersection of information security, enterprise collaboration, and AI. This role is pivotal in shaping how NVIDIA protects sensitive content, manages enterprise data access, and enables next-generation AI agents at scale. The platforms encompass sensitive information protection, a central data governance layer for enterprise content, and an AI knowledge platform for reliable AI agent consumption. The focus is on access control, export-control enforcement, and production readiness. This is a high-impact individual contributor role with potential for engineering leadership and management, requiring strong technical skills and the ability to influence stakeholders across Security, Legal, Finance, and AI teams.

Salary not listed

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 seeking a Senior High Performance AI Engineer to develop groundbreaking multi-agent systems within the CUDA ecosystem. This role involves building innovative agentic runtimes and compiler-integrated orchestration that leverage NVIDIA's software stack to accelerate modern agent workloads powered by foundational models. You will contribute to new agent abstractions, GPU-centric runtimes, and system solutions that enhance agent planning, tool-use, and code generation. The position requires close collaboration with internal NVIDIA software and hardware teams to integrate cutting-edge developments into NVIDIA products.

$184k - $357k

NVIDIA is seeking an AI Hardware Architect to design and analyze the next generation of Artificial Intelligence hardware, focusing on building innovative AI chips and chiplets for the AI infrastructure market. This role involves understanding computer architecture, microarchitecture, and chip design, with opportunities for continuous learning and growth. You will work on groundbreaking GPU and CPU systems, identify areas for improvement, and study the architectural implications of AI applications and models running on NVIDIA hardware.

$184k - $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

We are seeking a Senior Engineer to design, build, and scale the infrastructure that powers NVIDIA's AI agent ecosystem. This role sits at the intersection of distributed systems, developer platforms, and agentic AI, focusing on building foundational services that enable teams across the company to develop, deploy, orchestrate, and operate autonomous AI agents at production scale. You will be instrumental in creating the core platform that supports the full agent lifecycle, from registration to teardown, ensuring robust and scalable operations.

$200k - $391k

This role focuses on establishing and implementing a comprehensive evaluation strategy for autonomous agents at NVIDIA. The goal is to ensure that changes to agents, their underlying models, harnesses, tools, and skills lead to improvements rather than regressions. You will design systems that allow agents to leverage evaluation results for self-improvement, building trust in the process of autonomous evolution. This involves creating robust evaluation frameworks, mining production data for insights, and developing realistic simulation environments for safe, end-to-end agent task execution.

$224k - $431k

NVIDIA is seeking a highly motivated senior software engineer to join its communication libraries and network software team. This role involves developing and maintaining software for complex heterogeneous computing systems that power disruptive products in High Performance Computing and Deep Learning. You will be instrumental in designing and implementing highly-optimized communication runtimes and system software that enables interactions among GPUs and with other system components, contributing to the advancement of AI and HPC.

$152k - $288k

NVIDIA is seeking a motivated Performance Engineer to join our GPU Communications Libraries and Networking team. This role involves influencing the roadmap of communication libraries that are critical for Deep Learning and High Performance Computing applications running on massive GPU clusters. You will analyze and optimize communication performance between GPUs, which directly impacts end-to-end application performance at scale. This is an excellent opportunity to advance the state of the art in GPU communication and contribute to NVIDIA's innovative technologies.

$152k - $288k

NVIDIA is seeking a Senior Systems Software Engineer to focus on GPU Performance at Scale. In this role, you will drive innovation in AI and GPU computing by contributing to world-class computing hardware and software. You will provide insights into large-scale system composition and tuning for high-performance compute runs, collaborating with researchers, developers, and customers to create improved workflows and develop new solutions. You will also engage with HPC, OS, CPU, GPU compute, and systems specialists to architect, build, and optimize large-scale performance platforms.

$184k - $357k

Harvey is transforming legal and professional services by combining frontier agentic AI, an enterprise-grade platform, and deep domain expertise. This role is an opportunity to join a fast-scaling company at an inflection point, defining a new category in real-time. The team operates with intensity, ownership, and a commitment to excellence, guided by values of Decisiveness, Simplicity, and Job's Not Finished. We are looking for driven individuals to do the best work of their careers alongside like-minded colleagues.

$230k - $340k

Snorkel AI

Snorkel AI is seeking Senior and Staff Product Engineers with strong backend experience to help shape and deliver product capabilities that simplify complex workflows for users. This role involves owning backend features from concept to production, collaborating closely with Product, Design, and Engineering teams. You will be instrumental in building and evolving APIs, services, data models, and workflow capabilities that support the product, focusing on areas like data cataloging, agent benchmarking, and workflow automation. The ideal candidate possesses sound technical judgment and a commitment to product quality, usability, and impact.

$208k - $315k

SpaceXAI is seeking an engineer to join our RL infrastructure team and help develop our evaluation infrastructure. This role involves building highly reliable, efficient, and user-friendly systems to run evaluations at scale. You will collaborate with modeling and evaluation teams to develop new evaluations, maintain signal quality, and create internal tooling for training next-generation models. A key part of this role is identifying and resolving performance bottlenecks across the entire evaluation infrastructure stack, including inference, capacity fleet management, and asynchronous evaluation orchestration.

$180k - $440k

Crusoe

Crusoe is seeking a Senior Production Engineer to join their team and ensure the reliability and scalability of their AI-optimized cloud platform. This role is crucial for building and operating managed AI services at scale, focusing on delivering highly available, performant, and cost-efficient AI infrastructure for compute-intensive, latency-sensitive workloads. The ideal candidate will have a strong background in distributed systems and hands-on experience with large language models, contributing to the future of AI infrastructure.

$170k - $205k

Anthropic

As a Senior Software Engineer on the Life Sciences team, you will build and maintain biology software for use by Anthropic's scientists and the wider research community. This role involves accelerating biological progress with AI, combining frontier models with hands-on research. You will be responsible for the quality and reliability of released tools, pipelines, datasets, and benchmarks in genomics and structural biology, ensuring they are usable, trustworthy, and well-documented for external researchers. You will also represent the team within the computational biology community, contributing to public repositories, answering issues, and speaking at conferences. The role emphasizes using AI coding agents heavily in the software development lifecycle, holding agent-generated code to the same high standards as manually written code.

$300k - $320k

HackerRank is seeking a Senior Backend Software Development Engineer to drive the technical direction of critical backend systems and lead high-impact, platform-level initiatives. This role involves owning the architecture and evolution of core backend services, mentoring engineers, and making key technical decisions to ensure the platform remains fast, reliable, and scalable for a growing global user base. The ideal candidate will contribute to the company's mission of helping businesses hire and upskill developers based on skills, adapting to the evolving landscape where humans and AI build together.

$150k - $172k

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