Machine Learning Engineer Jobs in San Francisco

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 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 Senior Software Engineer to specialize in Deep Learning Inference. This role involves designing, building, and optimizing GPU-accelerated software for AI applications, focusing on high-performance deep learning frameworks like SGLang and vLLM for efficient large-scale model serving and inference. You will contribute to improving these platforms, enabling the smooth deployment of advanced language models and working with the deep learning community to implement and release the latest algorithms. The work will concentrate on enhancing the performance of state-of-the-art LLM and Generative AI models across various NVIDIA accelerators, utilizing open-source tools and kernels for optimization.

$184k - $357k

We are seeking senior engineers to contribute to the CUDA driver, a fundamental component for accelerating general-purpose computation on GPUs. This role involves enhancing features and improving performance to maximize NVIDIA hardware potential across diverse workloads, including deep learning, scientific computation, gaming, and virtual reality. The position requires strong system software programming, a deep understanding of operating systems and kernel programming, hardware architecture, and excellent communication skills. You will collaborate with hardware and software engineers to design, develop, debug, and deploy critical aspects of NVIDIA hardware and mobile system-on-chip devices. The work spans the entire development lifecycle, from design feedback and pre-silicon simulation to silicon bring-up and production software deployment.

$152k - $288k

NVIDIA is seeking a senior applications engineer with a background in material science codes to accelerate AI-driven simulations for materials discovery. This role will contribute to the development of NVIDIA ALCHEMI, a key software pillar for the next era of scientific discovery, bridging traditional simulations and AI surrogates. You will transform cutting-edge research into practical software tools for scientists, working collaboratively across research, engineering, and product teams.

$168k - $322k

NVIDIA is at the forefront of the AI revolution, exploring the next frontier of Physical AI, which involves machines that can perceive, reason about, and act within the three-dimensional world. The NVIDIA Cosmos platform provides open foundation models for Physical AI, designed to interpret images, video, and text to create a structured understanding of physical scenes, including motion, object interactions, geometry, and physical context. These models serve as the reasoning layer for robots, autonomous vehicles, and smart infrastructure. The Cosmos Engineering team is responsible for developing the core capabilities behind these models and is seeking a Senior Deep Learning Engineer to lead the data engine and end-to-end training verification for 3D spatial reasoning and perception. This role involves ensuring models accurately learn and represent 3D spatial concepts, working closely with research scientists to translate hypotheses into measurable capabilities in shipped models, and focusing on the critical role of data and evaluations in achieving frontier model quality.

Salary not listed

We are seeking a skilled engineer to join our Autonomous Vehicles team, focusing on the L3/L4 autonomous-driving solution. You will be instrumental in developing a comprehensive, learned 3D/4D world model by fusing navigation, ego-motion, perception, and sensor signals. This role involves close collaboration with perception, prediction, planning, and simulation teams to create a world representation that is complete, temporally consistent, uncertainty-aware, and robust for challenging driving scenarios. Our vision is to develop a shared multimodal scene representation that supports road understanding, 3D object perception, motion prediction, and planning, aiming to achieve human-level AI for navigation in unconstrained environments.

Salary not listed

NVIDIA is seeking a leader in AI-enabled drug discovery science to coordinate research and development efforts. This role bridges RNA therapeutics, experimental data generation, computational modeling, and translational science. The ideal candidate is a hands-on researcher experienced in guiding scientific teams, making technical trade-offs, and ensuring alignment across experimental, data, model, and compute activities. You will be responsible for converting scientific objectives into executable plans, leading workstreams in data curation, machine learning, and platform engineering, and guiding closed-loop learning processes. This position requires establishing robust science-stream governance and providing clear reporting on progress, dependencies, and risks to program leadership.

$224k - $431k

NVIDIA is seeking a world-class computer architect to contribute to the development of future high-performance computing systems, with a focus on enhancing the power-constrained performance of the hardware. Ideal candidates will have a strong track record of understanding and analyzing memory systems architecture to improve performance per watt (perf/W) and performance per millimeter (perf/mm). A broad perspective across the field of computer architecture and depth in the area of power, performance, and area (PPA) analysis is highly desirable. NVIDIA has pioneered programmable GPUs and the CUDA language and is a world leader in high-performance computing technology, with aggressive plans for future processors. This position offers the opportunity to have a real impact in a fast-moving, technology-focused company.

$184k - $357k

NVIDIA is seeking an outstanding individual to join our platform SWQA team as a Platform Software Validation Engineer (SDET). This role involves developing and executing test plans for NVIDIA HGX/DGX/MGX platforms, covering servers, OS, firmware, and CUDA software stacks. You will be responsible for installing and testing various systems, OS, server firmware, and software stacks, and driving root cause analysis for reliability and validation test failures. The position requires building and developing server and OS level automation frameworks and tests, reviewing partner and supplier test results, and managing the bug lifecycle within an agile development team.

$140k - $270k

NVIDIA is seeking a dedicated and creative Security Engineer to join our GPU System Software team. In this role, you will be instrumental in developing mitigations, conducting security research, and building advanced security platforms and frameworks to identify vulnerabilities in complex systems software. This is an exceptional opportunity to contribute to the security of foundational technologies in gaming, professional graphics, and cloud computing, as well as AI, data centers, and healthcare. You will be part of an inclusive team focused on maintaining an industry-leading security posture for groundbreaking products.

$152k - $288k

Joining NVIDIA's DGX Cloud AI Efficiency Team means advancing the performance, efficiency, and resiliency of large-scale AI workloads. You will help AI researchers and platform teams understand end-to-end behavior across GPUs, networking, storage, and software stacks. Your work will shape scalable DGX Cloud systems, turn complex measurements into prioritized engineering decisions, and continuously raise the performance and reliability of AI workloads. Join a technically diverse team of infrastructure experts to unlock more efficient AI at scale.

$224k - $431k

Join NVIDIA's CAD Infrastructure team to build a next-generation software platform for semiconductor design. This role involves applying innovative graph neural network infrastructure to process massive graphs and contributing to every phase from architecture to implementation. You will compose the future of computing and make a lasting impact on the world by developing advanced tools and systems.

$168k - $311k

NVIDIA is seeking a Principal Perception Engineer to lead the design and productization of next-generation autonomous driving perception stacks. This senior individual contributor role involves setting technical direction for 3D obstacle perception, driving cross-functional execution, and mentoring engineers. The role requires deep hands-on involvement in architecture, algorithms, and implementation, with a focus on modern transformer-based, multi-modal, and vision-language techniques. The goal is to build intelligent machines that can perceive, reason, and solve complex problems using GPU-accelerated deep learning.

$272k - $431k

NVIDIA is seeking a Senior Perception Engineer to design and productize the next generation of autonomous driving perception systems. This role focuses on the core 3D obstacle perception pipeline, involving architecture and algorithm design, with a hands-on approach to implementation using modern transformer-based, multi-modal, and vision-language techniques. You will contribute to the development of state-of-the-art deep learning models for obstacle detection and tracking, leveraging multi-camera and multi-sensor fusion. The position requires defining objectives, prototyping architectures, running experiments, and adhering to best practices for training and evaluation, including techniques like large-scale pretraining and parameter-efficient fine-tuning. You will also help define and maintain performance metrics, analyze large datasets to identify failure modes, and contribute to the data strategy for perception, including model-assisted workflows and model-in-the-loop tooling. Collaboration with safety, systems, and software teams is essential to ensure solutions meet product requirements for safety, latency, resource usage, and robustness for large-scale deployment.

$184k - $357k

NVIDIA is seeking a Senior Radar Perception Engineer to contribute to the design and productization of their next-generation autonomous driving perception stack. This role involves working on the core 3D radar and multi-modal obstacle perception pipeline, focusing on architecture and algorithm design. The engineer will be hands-on with implementation, including modern transformer-based, radar-centric foundation models, and multi-sensor fusion techniques.

$224k - $357k

NVIDIA is seeking a Senior Applied Power Architect to join our Applied Power Architecture team. You will be instrumental in developing state-of-the-art GPUs for AI, Automotive, GeForce, and Mobile products. This role involves inventing methodologies, features, and techniques to tackle electrical, speed, and power challenges for next-generation AI architectures. You will drive power management initiatives to optimize efficiency and performance while adhering to electrical, thermal, and power constraints. The position requires a deep understanding of process characteristics and their impact on voltage-frequency (V-F) tradeoffs, as well as investigating multi-die technologies and proposing innovative solutions for packaging, interconnect, physical design, electrical, and thermal challenges. Collaboration with cross-functional teams, including ASIC designers and packaging experts, is essential for formulating effective solutions.

$168k - $311k

NVIDIA's Math Libraries team is seeking a senior engineer to contribute to kernel generation for AI and HPC, focusing on matrix operations, JITing, and fusions. This role involves developing GPU-accelerated mathematical libraries that power advancements in AI, scientific simulations, and data analytics across various industries. You will be instrumental in enabling leading organizations to revolutionize fields like healthcare, NLP, VR, and autonomous vehicles.

$184k - $357k

NVIDIA is at the forefront of visual and AI computing, driven by our pioneering work with GPUs. We are seeking a GPU PCIe and Boot Architect to join our team and design the architecture and infrastructure for critical system functions including boot, PCIe/CXL enumeration, and capability discovery. This role involves collaborating with global software, hardware, and verification teams to shape the next generation of GPUs, creating functional and performance models, and supporting post-silicon validation. If you are creative, autonomous, and passionate about building the industry's highest-performance chips, we encourage you to apply.

$108k - $196k

NVIDIA is seeking software engineers for the CUDA Tile team to support a new tile-based programming model for NVIDIA LPUs. This role involves designing and implementing compiler transformations, developing MLIR-based dialects and lowering passes, and optimizing the performance of tile-based kernels for various NVIDIA GPU and LPU architectures. The scope includes defining public APIs, crafting optimization techniques, and general software engineering. NVIDIA offers competitive salaries, equity, and a comprehensive benefits package, making it a desirable employer in the technology industry.

$152k - $242k

NVIDIA is seeking a Senior System Simulation Architect to drive the development of CPU architectures for AI, deep learning, high-performance computing, gaming, virtual reality, and autonomous vehicles. This role involves developing and integrating full-system functional models, bringing up system and application software in simulation and emulation, and supporting CPU architects and performance engineers. The position is crucial for pushing performance boundaries for NVIDIA's CPU products and integrating them seamlessly with NVIDIA's broader technology stack to enable faster AI model training, agentic use-cases, efficient data processing, and scalable cloud deployments.

$184k - $357k

NVIDIA is seeking a Senior MLOps Engineering Manager to lead the build, development, and operation of large-scale, end-to-end data and ML pipelines that power the company's autonomous driving products. This role involves leading a highly technical engineering team responsible for building and operating cloud-scale pipelines that ingest, validate, process, and transform multimodal sensor data into high-quality training, evaluation, and validation datasets. These pipelines are foundational to the autonomous vehicle program and directly enable customer-facing autonomy features. The ideal candidate will be a seasoned engineering leader with strong ownership and a passion for customer-focused development, capable of scaling systems and teams in a fast-paced, multi-functional environment.

$272k - $431k

NVIDIA is seeking a world-class computer architect to contribute to the development of future high-performance computing systems, with a focus on enhancing the power-constrained performance of hardware. The ideal candidate will have a strong track record of understanding and analyzing memory systems architecture to improve performance per watt and performance per millimeter. A broad perspective across computer architecture and depth in power, performance, and area (PPA) analysis is highly desirable. This position offers the opportunity to have a real impact in a fast-moving, technology-focused company.

$272k - $431k

We are the GPU Communications Libraries and Networking team at NVIDIA, responsible for delivering critical communication libraries like NCCL, NVSHMEM, and UCX for Deep Learning and High-Performance Computing (HPC) applications. These applications demand immense compute power and operate at scales involving tens of thousands of GPUs, connected via high-speed interconnects and networking. Communication performance is paramount, directly impacting end-to-end application performance, especially at massive scales. We are seeking a technical leader to manage our NVSHMEM and UCX libraries, offering an exceptional opportunity to advance the state-of-the-art and develop groundbreaking platforms.

$184k - $357k