AI Engineering Jobs in San Francisco

NVIDIA's TensorRT team is seeking talented engineers to advance high-performance AI inference on NVIDIA GPUs. This role focuses on developing software that accelerates state-of-the-art generative AI models for local and enterprise deployments. You will tackle challenging problems in GPU code generation, just-in-time compilation, and inference optimization to deliver high performance across diverse models and hardware configurations. As a Senior Software Engineer, you will design and implement advanced GPU inference software, translating a deep understanding of GPU execution into efficient code generation and measurable improvements in generative AI inference.

$152k - $288k

NVIDIA is seeking a Senior Compiler Engineer to join our GPU Software organization. In this role, you will be instrumental in delivering features and improvements that maximize the potential of NVIDIA hardware across diverse computational workloads, including deep learning, scientific computing, autonomous vehicles, and gaming. Our compiler organization plays a critical role in every GPU NVIDIA produces. You will focus on the CUDA compiler front end, developing language features, diagnostics, and capabilities that empower both human developers and AI agents to create correct and high-performance GPU code. This position offers the opportunity to expand your expertise into other areas of the compiler stack, such as the LLVM-based optimizer, code generation, and MLIR.

$152k - $242k

NVIDIA is seeking a Security Research Engineer to join its AI Safety & Security Engineering team. This role focuses on advancing AI-powered tooling to find, validate, and patch software vulnerabilities. You will apply rigorous security judgment to confirm vulnerabilities and define correct, safe fixes, ensuring they repair flaws, protect existing behavior, and withstand revalidation. The work involves developing repeatable methods, establishing high engineering standards, and documenting reasoning for security reviews. While AI-assisted workflows are integral, a disciplined skepticism towards their outputs is also crucial. Your contributions will shape the program's understanding of trustworthy patching and help decide future vulnerability class targets.

$224k - $431k

Our team builds AI-driven software systems for Circuit Design, combining automation algorithms, deep learning models, and agentic workflows to accelerate end-to-end design automation. We are seeking an Applied Machine Learning Engineer to join our integral team in the Circuit Solutions Group. You will work within a multi-functional team on various projects involving pre-silicon and post-silicon custom circuit design, circuit/layout optimization, and Spice correlation. This role involves researching and implementing frontier solutions in electronic design automation and building innovative agentic AI solutions for VLSI design problems. You will be responsible for analyzing problems and datasets, validating hypotheses, and designing and building models and algorithms to achieve desired quality of results.

$152k - $265k

NVIDIA is seeking a founding Director to establish and lead its Forward Deployed Engineering (FDE) organization for open models, specifically focusing on NVIDIA Nemotron and NVIDIA Cosmos across North America. The core mission is to transform open-source models into production-ready AI solutions tailored to specific domains, leveraging NVIDIA's partner ecosystem. This role involves setting the strategic direction for North America, building and scaling the FDE team, and fostering strong relationships with partners, customers, and NVIDIA leadership. The FDE team will work collaboratively with top partner engineering and research teams, co-leading joint customer engagements and developing best practices to ensure partner success and scalability. The ultimate goal is to empower organizations to build, own, and operate their AI on open-source models.

$320k - $489k

We are seeking a Senior Forward Deployed Engineer to lead technical engagements with partners and customers, focusing on NVIDIA's open model portfolio. The primary mission is to transform open models into production-ready AI solutions tailored to specific domains, covering the entire lifecycle from post-training to harness optimization. This role involves collaborating closely with NVIDIA's partner ecosystem, co-leading customer engagements, and developing resources that enable partners to scale their AI initiatives. You will work hands-on to build custom AI solutions, from initial customer conversations through to production deployment, ensuring models are optimized for performance and cost-effectiveness.

$184k - $357k

NVIDIA is seeking software engineers for the Tensor IR & CUDA Tile team to work on compiler technologies for NVIDIA GPUs. These technologies are central to the deep learning revolution, enabling breakthroughs in generative AI, large language models, and other critical areas. You will join a top-notch team with broad impact across the deep learning community, contributing to the next era of computing where AI drives innovation.

$152k - $242k

The DL Performance Modeling Team is seeking a Deep Learning Software Engineer to develop full-stack simulation infrastructure for deep learning applications across various GPUs. This role involves collaborating with cross-functional teams to shape the roadmap for DL hardware and software, and to rapidly assess new AI-accelerating GPU hardware and software advancements through simulation.

$152k - $288k

NVIDIA is seeking a Senior Software Engineering Manager to lead the development of a production-quality reinforcement learning (RL) ecosystem. This role involves building and extending RL post-training frameworks and the underlying systems that enable modern AI to reason, use tools, and act as agents. You will be responsible for setting strategy, building and managing a globally distributed team, and translating technical and customer signals into clear engineering priorities. The work spans various RL frameworks and their supporting systems, requiring deep systems expertise to ensure reliability and scalability across GPUs, CPUs, networking, and storage.

$272k - $431k

NVIDIA is seeking a Reinforcement Learning Engineer to join its mission in building intelligent, safe, and efficient self-driving technology. This role involves pushing the boundaries of autonomous vehicle decision-making and planning by leveraging Deep Reinforcement Learning (RL). As a world leader in AI and high-performance computing, NVIDIA offers a unique platform where innovative research meets real-world production, aiming to redefine transportation globally.

$224k - $357k

We are seeking exceptional Senior Machine Learning and Simulation Engineers to join NVIDIA's Autonomous Vehicles (AV) Simulation team. This role requires strong technical leadership and outstanding software engineering skills, coupled with deep expertise in both simulation and artificial intelligence, including deep learning, reinforcement learning, end-to-end driving and Physics AI models. The successful candidate will have a solid track record of productizing ML solutions for autonomous driving and simulation at scale. This position centers on developing a Closed-Loop Simulation-based Reinforcement Learning (RL) framework in order to train advanced end-to-end AV models, such as Alpamayo R1. This position will design and improve the accuracy and performance of the RL framework and simulation, leveraging SOTA techs including NuRec, Traffic Models, and Cosmos World Model. Success in this role requires close collaboration with the AV Platform, AV Product, and Research teams.

$224k - $431k

NVIDIA DRIVE is developing solutions for safe, scalable, AI-enabled autonomous vehicles, spanning in-car supercomputers to cloud-scale training. The team is seeking passionate, AI-powered engineers to join the DRIVE Mapping team. In this role, you will contribute to delivering efficient SD maps to in-car consumers, focusing on localization, routing, and navigation to enhance vehicle safety. If you are motivated by solving complex problems for self-driving cars and have a background in software design, embedded software, and real-time operating systems, we encourage you to apply.

$152k - $288k

NVIDIA is seeking a creative and expert Systems Engineer to join its Automotive Vehicles team and contribute to the expansion of NVIDIA's autonomous vehicle solutions. This role involves a focused effort on productizing groundbreaking perception solutions that will redefine transportation and the self-driving car industry. You will collaborate with a global team of systems engineers, AI applied researchers, and software engineers to build world-class perception, mapping, and localization solutions, ensuring adherence to all NVIDIA and automotive standards and guidelines. The work is described as exciting, fun, and relevant, offering an opportunity to make a lasting impact.

$184k - $357k

NVIDIA is building the next generation AI-native autonomous driving architecture, integrating classical safety stacks, foundation models, and scalable AI systems into a unified production platform. This role focuses on the future of autonomous systems, moving beyond purely classical robotics or end-to-end neural networks to a deep integration of safety architectures and large-scale AI-driven systems. If you are passionate about shaping the future of autonomous systems over the next decade, this is an exceptional opportunity.

$184k - $357k

NVIDIA is seeking a creative and hands-on automation engineer with strong Python development skills and a quality-focused attitude to join our team. This role will focus on improving software quality through thoughtful testing, building and maintaining Python-based automation frameworks, and utilizing AI development tools to accelerate test creation and implementation. You will contribute to both manual and automated test planning and execution for NVIDIA App, a key product in our software suite.

$140k - $270k

NVIDIA is seeking a Senior Software Engineer to advance the C++ foundation of CUDA Core Libraries. This role involves designing and optimizing high-performance algorithms and APIs for C++ developers, contributing to the foundational libraries, algorithms, and language/runtime infrastructure that power CUDA. You will be responsible for the full lifecycle of features, from design and implementation to testing, profiling, and maintenance, ensuring a speed-of-light experience for developers and AI coding agents.

$184k - $357k

NVIDIA is seeking a Senior Software Engineer to enhance the Rust experience for CUDA Core Libraries. This role involves designing and implementing safe, idiomatic Rust APIs for GPU computing, while integrating closely with existing C/C++ components. You will be part of the team responsible for building the foundational libraries, algorithms, and infrastructure that power CUDA, ensuring a high-performance experience for developers and AI coding agents.

$184k - $357k

NVIDIA is seeking software engineers to work on the CUDA driver for Windows. CUDA is NVIDIA's 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, from deep learning and scientific computation to video games and virtual reality. 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 shape the future direction of CUDA as you collaborate with your peers across NVIDIA.

$184k - $357k

NVIDIA is seeking outstanding senior engineers to work on the CUDA driver, a key component of accelerated GPU computing. You will join a versatile software engineering team that delivers innovative software features to unlock the full potential and performance of NVIDIA hardware across diverse workloads like deep learning, scientific research, autonomous vehicles, gaming, and virtual reality. This multi-functional role requires collaboration with hardware architects, deep learning specialists, and both internal and external partners to advance the CUDA architecture. With the opportunity to collaborate with teams across the whole NVIDIA computing stack, you will help design software solutions across kernel mode components, compilers, and networking software. Your system-level expertise and creativity in solving complex problems will help invent the future of CUDA and NVIDIA’s compute technologies! Does crafting solutions to influence the evolution of GPU-accelerated computing sound exciting? If so, we invite you to join us and help engineer the next wave of innovation at NVIDIA!

$152k - $288k

We are hiring software 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, and the operating system. Unified Memory kernel driver provides kernel memory management that enables these advanced features. As a member of our team, you will apply your design, coding expertise, and creativity, and to collaborate with peers across NVIDIA to deliver upon the best compute platform in the world. You will craft sophisticated solutions to exciting problems shaping the future direction of CUDA!

$184k - $357k

NVIDIA is seeking a Senior Software Engineer to enhance the Python experience for its foundational CUDA Core Libraries. This role involves building Pythonic APIs, language bindings, algorithms, and runtime infrastructure that sit on top of native C/C++ foundations. You will contribute to the core libraries and infrastructure that make CUDA a high-performance platform for developers and AI coding agents. The position requires a strong focus on designing and implementing efficient, reliable, and scalable GPU-accelerated software, with an emphasis on improving the Python developer experience through robust tooling and clear documentation.

$184k - $357k

NVIDIA is seeking a motivated Deep Learning engineer to integrate advanced CUDA features and Distributed Runtime technologies into AI stacks, including frameworks like PyTorch, TRT-LLM, vLLM, SGLang, and JAX. You will join the team responsible for core CUDA features and runtimes that scale Deep Learning and HPC applications, addressing diverse multi-GPU demands from large-scale training to low-latency inference. Your work will enhance the productivity and performance of AI applications, accelerating community adoption through AI toolkits. This role offers a significant opportunity to advance the state of the art in AI technologies.

$184k - $357k

NVIDIA is seeking a Machine Learning Engineer to join the GPU accelerated Apache Spark team. This role involves working with the open-source community to enhance Apache Spark's performance and cost-efficiency on NVIDIA GPUs, enabling enterprises to scale their Spark workloads. You will apply advanced ML/AI techniques to optimize these large-scale data processing applications.

$224k - $357k

NVIDIA is seeking a motivated system software engineer with deep understanding of device drivers and strong C/C++ skills to work on the CUDA Driver. This role is integral to a team that delivers features and improvements to realize the full potential of NVIDIA hardware for a wide range of computational workloads, including deep learning, scientific computation, data science, self-driving cars, video games, and virtual reality. You will use your design abilities, coding expertise, and creativity to craft elegant solutions to exciting problems and shape the future direction of CUDA by collaborating with peers across NVIDIA.

$152k - $288k