Senior AI Engineering Jobs

NVIDIA is seeking a Principal Engineer to drive the performance of large-scale AI training and post-training workloads across NVIDIA’s full hardware and software stack. This role sits at the intersection of distributed training, GPU architecture, systems software, deep learning frameworks, and performance engineering. You will analyze and optimize frontier-scale LLM workloads running on thousands of GPUs, drive improvements across frameworks such as PyTorch, JAX, NeMo, and NeMo RL, and use insights from real workloads to help shape future NVIDIA GPU, system, and software roadmaps. We are looking for a deeply technical leader who can operate across abstraction layers: from application-level training behavior to framework/runtime internals, CUDA libraries, communication collectives, memory systems, networking, and GPU architecture. At this level, success means both directly improving performance directly as well as setting technical direction, raising the bar for the organization, and influencing multi-functional decisions across NVIDIA.

$272k - $431k

NVIDIA is seeking a senior or principal engineer to join the Generalist Embodied Agent Research (GEAR) group, focusing on robotics systems. This role is central to Project GR00T, an initiative to build foundation models and full-stack technology for humanoid robots. You will collaborate with a research team known for its influential work in multimodal foundation models, large-scale robot learning, embodied AI, and physics simulation, contributing to impactful research projects and product roadmaps.

$184k - $357k

We are seeking a Senior Performance Modeling Architect to define and enhance our next-generation CPU Cache Hierarchies and interconnects. This role offers a unique opportunity to develop scalable solutions that bridge the high-reliability, low-latency requirements of Automotive systems with the efficiency and density demands of Data Center environments. You will be instrumental in creating the core models that manage data flow across our silicon, ensuring our L3/System Caches and coherent fabrics meet ambitious performance targets.

$152k - $288k

NVIDIA is seeking a GPU System Architect to design multi-GPU scale-up and scale-out systems for next-generation AI and HPC datacenter platforms. This role involves defining system architectures that integrate GPU compute, high-bandwidth memory, and interconnects to achieve industry-leading AI performance, scalability, and resilience. The ideal candidate will possess extensive experience in system-level fabric/networking architecture and hardware-software co-design.

$184k - $357k

NVIDIA is seeking a Senior AI Frameworks Engineer to join the CUTLASS team, a popular open-source ecosystem for high-performance math primitives on NVIDIA GPUs. This role focuses on developing Pythonic CUTLASS (CUTLASS DSL), aiming to bring high-performance abstractions directly into the Python environment. You will bridge the gap between low-level hardware primitives and high-level developer productivity, empowering AI researchers and engineers with better tools. If you are passionate about building elegant, high-performance DSLs, this is an opportunity to contribute to the next frontier of GPU programming.

$152k - $288k

NVIDIA is seeking an AI Compiler Engineer with deep expertise in compiler technologies to join their team. This role focuses on building AI-driven compiler intelligence for production compiler pipelines, aiming to deliver measurable improvements in performance and efficiency. The engineer will advance LLM-enabled workflows for compiler decisioning and developer productivity, working within a diverse and supportive environment that fosters innovation. The position involves collaborating with world-class engineers to shape next-generation compiler capabilities that power large-scale, high-impact products.

$184k - $288k

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 is seeking a Senior Applied Research Scientist with deep quantum computing expertise to pioneer the future of fault-tolerant quantum systems powered by machine learning. This role involves architecting and building AI solutions at the core of quantum error correction, decoding, and calibration. You will research and develop open AI models, curated datasets, and rigorous benchmarks to advance the state of the art and empower the quantum community. Your work will translate cutting-edge theory into practice by fine-tuning models for specific quantum error-correcting codes and hardware platforms, collaborating with multi-functional teams to integrate AI into next-generation Accelerated Quantum Supercomputers.

$192k - $305k

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

NVIDIA is seeking a senior software engineer for its AI Networking Codesign and Benchmarking R&D group. In this role, you will be responsible for profiling, analyzing, and optimizing AI workloads on large-scale GPU and CPU clusters, focusing on distributed Deep Learning LLM training and inference. Your work will center on collective communication and networking, involving collaboration across hardware components like HCAs, Switches, CPUs, and GPUs, as well as software layers including LLM applications, machine learning frameworks, and communication libraries. You will also develop performance analysis tools and strategies to identify and clarify performance expectations, limitations, and bottlenecks, contributing directly to AI innovation.

$272k - $489k

NVIDIA is seeking dedicated Software Engineers to join its custom RAM design team, pushing the boundaries of memory technology for AI and accelerated computing. This role involves developing and deploying AI agents that leverage machine learning models to solve complex problems in VLSI design. You will work closely with experts in VLSI design, machine learning, and software engineering to create and integrate ML-based solutions that enhance the efficiency, productivity, and quality of VLSI design flows. This is an opportunity to make a lasting impact on the world by applying cutting-edge AI technologies to chip design.

$136k - $265k

NVIDIA is seeking a Senior Scientist to advance synthetic data generation for training frontier models. You will contribute to open-source libraries within the NVIDIA NeMo ecosystem, creating synthetic datasets across text, code, structured, and multimodal data to enhance LLM pre- and post-training. This role blends hands-on software engineering with applied research in generative methods, requiring collaboration with research, engineering, product, and model teams, as well as external labs. The position emphasizes driving software excellence through modern tooling and professional Git/CI-CD practices, and publishing original research to maintain technical leadership.

$168k - $305k

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

NVIDIA seeks a Senior Deep Learning Hardware Modeling Architect to join a team focused on pushing the performance boundaries of AI inference through ambitious hardware-software co-design. This role offers a unique opportunity to model critical components of world-class LLM inference solutions, contributing to the state of the art in LLM inference hardware. The ideal candidate is passionate about this domain and possesses a proven record of C++ hardware modeling experience.

$152k - $288k

NVIDIA is seeking a senior software engineer to accelerate quantum chemistry on NVIDIA GPUs with cuEST, a CUDA-X math library. This role involves expanding the frontiers of electronic structure theory for applications like drug discovery and semiconductor development. You will be involved in the day-to-day engineering of high-performance math libraries and innovating new algorithms to enhance practical quantum chemistry.

$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

NVIDIA is seeking a Senior SoC Subsystem & I/O Architect to join its LPU team. This role is responsible for defining the high-level architecture for next-generation AI and high-performance computing products. The position requires deep expertise in SoC subsystems and uncore IPs, including interconnects, memory systems, coherency, boot processes, system initialization, protection mechanisms, reliability, power management, and debug interfaces. The architect will translate product requirements into detailed architectural specifications and collaborate closely with IP teams to develop build documents. This role involves building and utilizing models to validate architectural intent and subsystem behavior before RTL or silicon development.

$184k - $357k

NVIDIA is seeking experienced ASIC Design Engineers to join our "AI for chip design" team, focusing on agentic AI to revolutionize the design of world-leading SoCs and GPUs. This role offers a unique opportunity to leverage the power of AI to transform chip design at NVIDIA and across the industry. You will be part of a global team pushing the boundaries of what's possible in computing, working at the intersection of research, engineering, and product development to turn innovative ideas into real-world solutions. If you are passionate about generative AI and cutting-edge technologies, this is your chance to be at the forefront of innovation.

$168k - $311k

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

At NVIDIA OpenShell, we are building the runtime infrastructure for secure, scalable, production-grade AI agents. As an OpenShell Research Engineer, you will help bring the latest advances in agentic systems into the runtime, tools, and workflows that enterprise builders rely on. This role sits at the intersection of research, product, and engineering, requiring you to identify promising methods from academia and industry, validate their impact through prototypes and benchmarks, and help integrate the best ideas into the product.

$184k - $288k

NVIDIA is seeking a Senior Research Engineer passionate about Generative AI inference to help revolutionize how AI is integrated into products and services. Our team is at the forefront of developing optimized inferencing technologies for a wide range of generative AI models, from language to images. We are involved in the entire machine learning lifecycle, from initial concepts and applied research to engineering for optimized inference and final deployment. This role involves close collaboration with research teams, engineers, and the open-source community.

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