Machine Learning Engineer Jobs in San Francisco

NVIDIA BioNeMo is seeking a Senior Software Engineer to join the cuEquivariance team. This team develops and ships production GPU kernels and software interfaces that accelerate geometric neural networks on NVIDIA GPUs, enabling researchers in fields like molecular biology, materials science, and physics to train and deploy equivariant models at scale. The role involves CUDA kernel engineering, Python library development with PyTorch and JAX, and collaboration with research teams and external framework developers. This is an opportunity to work at the intersection of GPU computing and graph-based deep learning, with your work impacting scientific communities through production pipelines.

$184k - $357k

NVIDIA's CUTLASS is a popular open-source ecosystem for high-performance linear algebra and Tensor Core primitives, providing C++ and Python abstractions for computations on NVIDIA GPUs. This role focuses on bridging the gap between current and theoretical performance, optimizing GPU kernels and fusion opportunities within deep learning models. You will develop tooling for automated benchmarking and analysis, becoming an authoritative resource on kernel performance and collaborating with various NVIDIA teams.

$152k - $288k

We are seeking software engineers to join our development efforts on cuDSS, a CUDA library for direct solvers of sparse linear systems. Our team develops GPU-accelerated libraries and SDKs that power scientific and engineering simulations, data analytics, and AI across various industries. In this role, you will collaborate with other developers to craft algorithms and kernels for direct sparse solvers, contributing to the advancement of accelerated computing.

$184k - $357k

The Graphics Developer Tools team is seeking an exceptional technical leader to shape the future of graphics development. This role involves driving innovation across developer tools, graphics technologies, and next-generation development workflows within NVIDIA's graphics software ecosystem. You will lead new strategic initiatives, collaborating with NVIDIA Research, GPU Architecture, Driver, SDK, and Developer Technology teams to evaluate emerging technologies, prototype ideas, and deliver solutions that enhance how developers build, debug, profile, and optimize graphics applications. This is a hands-on technical leadership position requiring vision, execution, influence, and collaboration, where success is achieved through technical excellence and aligning teams around ambitious goals.

$272k - $431k

Join the NVIDIA GeForce NOW cloud team, responsible for delivering high-quality PC gaming experiences across various devices. This role focuses on building diagnostic, prescriptive, and AI-augmented analytics solutions, including real-time demand forecasting, capacity allocation, anomaly detection, and customer analytics. You will leverage data and AI to enhance cloud computing and streaming performance for millions of users, working with a technology stack that includes Python, Spark, SQL, Databricks, MLFlow, and Kubernetes.

$248k - $380k

NVIDIA's LPX System Software team is seeking a Principal Software Engineer to build foundational software for a novel deterministic compute architecture. This role involves developing hardware abstraction layers, core system libraries, drivers, and runtime components, with a strong emphasis on memory safety and API stability using Rust. The position requires setting technical direction, shaping architecture, and driving complex reliability and bring-up challenges. A key aspect of this role involves leveraging AI coding agents as a primary workflow tool, requiring senior engineers to be adept at directing them and designing systems that are legible to both humans and AI.

$272k - $431k

NVIDIA's Metropolis team is at the forefront of transforming how the physical world is perceived and understood through advanced computer vision and deep learning. We build large-scale distributed Vision AI platforms that power intelligent spaces, smart cities, retail analytics, and digital twins. This role offers the opportunity to own core components of a strategic platform with high visibility and real-world impact. As a System Software Engineer for Vision AI, you will develop and optimize high-performance vision systems that turn massive streams of video, image, and 3D data into actionable insights. You will collaborate with specialists in perception, simulation, and large models to bring research into production at scale.

$224k - $357k

NVIDIA is seeking an exceptional engineering leader and manager to head up the Isaac for Manipulation team. This role focuses on giving robotic arms and dexterous systems the perception, grasping, and motion skills needed for real-world applications. You will be responsible for the strategy, roadmap, and execution for manipulation capabilities, addressing complex contact-rich problems in manufacturing, logistics, and industrial automation. This is a high-visibility position with the opportunity to grow and mentor a world-class team, set technical direction, and partner with the robotics ecosystem to establish NVIDIA as the leading platform for robotic manipulation.

$272k - $431k

NVIDIA's FlashDreams platform is at the forefront of developing the next generation of interactive world models. We collaborate with leading model makers and early adopters in fields like autonomous vehicles, robotics, medical training, and gaming to bring these advanced technologies into production. This role bridges research and systems engineering, focusing on transforming innovative ideas into robust, high-performance systems for both developers and customers. If you are passionate about advancing interactive world models from research concepts to real-world applications, we encourage you to apply.

$184k - $357k

NVIDIA is seeking a Senior Staff Engineer to lead the invention, reliability, security, and evolution of global employee communications and collaboration platforms. This role focuses on deep ownership of Microsoft Exchange (on-premises, hybrid, and Exchange Online/O365), Azure-based services, and SMTP/IMAP integrations, while also encompassing broader enterprise SaaS ecosystems like Slack, Microsoft 365, SharePoint, and Google Workspace. The ideal candidate will possess strong Exchange expertise and a desire to expand their impact across a modern, AI-augmented enterprise collaboration stack, ensuring NVIDIA teams can communicate, collaborate, and deliver outstanding products effectively.

$168k - $311k

NVIDIA is seeking a senior engineer obsessed with performance analysis and optimization to help squeeze every last clock cycle out of AI training, the workload driving the design and construction of the largest and most powerful compute systems in the world. This role offers the opportunity to directly impact the hardware and software roadmap in a fast-growing technology company that leads the AI revolution. Join us and help design and build the world's most powerful compute systems!

$184k - $357k

NVIDIA's Metropolis team is advancing physical AI by building intelligent video analytics and perception solutions for smart cities, industrial automation, and autonomous systems. We are seeking a Senior ML Engineer to develop and deliver next-generation Metropolis solutions powered by Cosmos, NVIDIA's world foundation model platform. This role involves constructing production-quality AI capabilities for Metropolis by demonstrating Cosmos to address real-world perception and physical AI challenges. You will also collaborate with the Cosmos team to enhance the platform's features in line with the Metropolis product roadmap. If you are passionate about pushing generative AI, simulation, and large-scale model development forward and want your work to impact real-world systems, we encourage you to apply.

$184k - $357k

NVIDIA is seeking a dedicated Software Engineer, ML Performance to join our team in Santa Clara, CA. In this role, you will engage directly with the latest AI technologies and work on optimizing performance at an outstanding scale. Your expertise will ensure that our groundbreaking solutions operate flawlessly and continue to lead the industry. You will be immersed in an inclusive, supportive environment where everyone is inspired to do their best work. Come join the team and see how you can make a lasting impact on the world.

$224k - $357k

NVIDIA is seeking an NCX Senior Engineer to join our DSX team, collaborating closely with strategic customers to implement and enhance groundbreaking AI workloads. You will deliver hands-on technical assistance for advanced AI deployments, intricate distributed systems, and ensure customers realize efficient performance from NVIDIA's AI platform across varied environments. We partner with the world's most innovative AI companies to address their most challenging technical problems. In this role, you will develop innovative solutions that advance AI infrastructure capabilities and directly influence customer success with breakthrough AI initiatives.

$184k - $357k

NVIDIA is seeking a creative, hands-on software engineer with a test-to-failure approach to join our team. You will be responsible for building data-driven tools to improve software quality and ensure customers have the best experience with GeForce NVIDIA products. This role involves understanding software and hardware specifications, building reliable tests and tools in C++/C#/Python, and leveraging AI technologies for test case automation. You will also gain in-depth understanding of both Windows and Linux operating systems.

$140k - $270k

NVIDIA is seeking an experienced software engineer to help build NVIDIA AI for Media SDKs, a GPU-accelerated platform delivering AI-powered video, audio, and augmented reality capabilities for developers building media, content creation, and broadcast workflows. You will develop and optimize AI software across cloud and embedded platforms while advancing the build, validation, and delivery infrastructure behind our SDKs. This role combines strong C++ systems engineering, Linux expertise, accelerated inference, and software delivery automation. You will collaborate with algorithm, platform, infrastructure, and QA teams to deliver efficient, reliable software on NVIDIA hardware.

$184k - $357k

At NVIDIA, we are redefining the future of computing. As a Senior Verification Engineer on our CPU Verification Team, you will play a pivotal role in our mission to push the frontiers of technology. Your work here will impact a broad array of products, from consumer graphics to self-driving vehicles and the burgeoning field of AI. This is an outstanding chance to create a real difference within a company that values innovation, collaboration, and excellence.

$168k - $311k

NVIDIA GPU Architecture Group is seeking a senior software engineer to automate and optimize performance analysis workflows for AI training and inference workloads. You will not only perform analysis but also reshape how it's done, building tools and workflows that scale across a broad audience of engineers. NVIDIA is a world-class leader in AI workload optimization, and your work will push these boundaries even further!

$184k - $357k

We are seeking a Senior DFX Software Engineer with a focus on Machine Learning to join our innovative team. This role involves defining and building methodologies, software, and flows for silicon device testing, debug, and failure analysis. You will contribute to advancing NVIDIA's participation in fields like gaming, compute platforms, and Artificial Intelligence by enabling high-quality silicon defect screening. If you are a creative, problem-solving individual who enjoys tackling technical challenges with new approaches, we encourage you to apply.

$152k - $288k

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

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

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