Back to feed
News Story
APriority84
NVIDIA Developer Blog
1 sources

Inside NVIDIA Rubin GPU Architecture: Powering the Era of Agentic AI

NVIDIA's new Rubin GPU architecture is designed to power the next generation of agentic AI systems, enabling always-on AI factories that produce intelligence at scale. This marks a shift from discrete training and chat interfaces to continuous, autonomous AI operations.

SynthePulse Insight · AI deep reading

NVIDIA Rubin GPU: Reshaping Architecture for Agentic AI Inference

Version 1 · 1 source

The NVIDIA Rubin GPU, with 336 billion transistors, HBM4, and a third-generation Transformer Engine, claims a 10x improvement in agentic inference energy efficiency over Blackwell. This article, based on the official technical blog, breaks down its architectural innovations and key constraints.

  • Rubin GPU integrates 336 billion transistors, 224 SMs, 896 Tensor Cores, supports NVFP4 precision, and delivers peak performance of 50 petaflops.
  • Equipped with 288 GB HBM4 memory with 22 TB/s bandwidth; NVLink 6 provides 3600 GB/s inter-chip interconnect.
  • Third-generation Transformer Engine supports dynamic precision adjustment, claiming improved inference efficiency while maintaining accuracy.
  • Enhanced Tensor Memory Accelerator (TMA) supports inline descriptor updates, optimizing MoE model weight loading.
  • Tensor Core doubles instruction throughput in the K dimension, reducing GEMM loop iterations and improving matrix operation efficiency.
  • Vera Rubin NVL72 rack-scale solution integrates liquid cooling, power smoothing, cable-free MGX architecture, supporting models with trillions of parameters.
Open section navigationCore Architecture: Transistor Stacking and Memory Revolution

Core Architecture: Transistor Stacking and Memory Revolution

The Rubin GPU is composed of two compute chips interconnected via NV-HBI high-speed packaging, totaling 336 billion transistors, with 224 streaming multiprocessors (SMs) and 896 Tensor Cores. Its third-generation Transformer Engine supports NVFP4 precision, officially claiming up to 50 petaflops of inference performance while maintaining model accuracy.

The memory subsystem uses 12-Hi stacked HBM4, with 288 GB capacity and 22 TB/s peak bandwidth. For inter-chip connectivity, NVLink 6 provides 3600 GB/s bandwidth, NVLink-C2C achieves 1800 GB/s CPU-GPU coherent interconnect, and PCIe Gen 6 x16 offers 256 GB/s host connection. Additionally, TEE-I/O confidential computing is used to protect data throughout its lifecycle.

MoE and Matrix Operations: Reducing Data Movement Overhead

For mixture-of-experts (MoE) models, Rubin enhances the Tensor Memory Accelerator (TMA) to support inline descriptor updates. Traditional approaches require modifying descriptors in memory, while Rubin allows kernels to directly overwrite pointer and stride fields in TMA instructions, reducing metadata management overhead and improving expert weight loading efficiency.

Tensor Core instruction throughput in the K dimension is doubled, halving the number of GEMM loop iterations. For example, Blackwell requires 4 K iterations for an operation, while Rubin needs only 2. This reduces loop overhead and improves Tensor Core utilization, especially beneficial for context and decode GEMM operations with large tensor parallelism.

Long Context and Agentic Workload Challenges

Agentic AI workloads require sustained inference, multi-step planning, and long-context attention, placing higher demands on KV cache capacity and low-latency decoding. Rubin addresses this with a large L2 cache and GigaThread engine for task scheduling coordination, but the official blog does not provide specific long-context performance data or benchmarks compared to Blackwell.

The rack-scale Vera Rubin NVL72 integrates liquid cooling, power smoothing (DSX MaxLPS), cable-free MGX architecture, and hot-swappable NVLink switch trays, claiming support for up to 40% more GPUs at the same power consumption and enabling deployment of models with trillions of parameters.

Credibility boundary

All technical details in this article are from NVIDIA's official technical blog, representing first-party product promotion. Performance claims (e.g., 10x energy efficiency) are based on internal 2T MoE workload tests, without third-party verification or standardized benchmarks. Some architectural improvements (e.g., TMA inline updates) are described in detail, but actual effectiveness awaits independent evaluation.

Insight takeaway

The Rubin GPU significantly improves transistor scale, memory bandwidth, and interconnect capabilities, with architectural optimizations for MoE and long-context inference. However, key metrics like the 10x energy efficiency improvement are based solely on internal tests and lack detailed comparison data with Blackwell; actual performance requires further verification.

Primary report

NVIDIA Developer Blog

Primary source