NVIDIA logo

How NVIDIA Uses PyTorch

618 engineering articles about PyTorch from NVIDIA's engineering team

Articles

Filter:
NVIDIA logo
NVIDIA
Advanced
As language models grow, scaling dense architectures becomes increasingly expensive. In a dense transformer, every token passes through every layer…
Michelle Horton
6 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
As large language model (LLM) inference increasingly processes sensitive information and proprietary model context across personal, enterprise…
Tanya Lenz
6 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
The compute and memory demands of generative AI increasingly exceed what a single GPU can provide. NVIDIA TensorRT multi-device inference is a new capability…
Tanya Lenz
6 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Weather-sensitive industries increasingly have access to observations that offer an earlier, more local view of changing conditions.
Elizabeth Goodman
13 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
cuTile Rust () is a tile-based system for safe, idiomatic GPU kernel authoring in the Rust programming language. Extending the Rust ownership model to tile…
Tanya Lenz
18 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Biomolecular structure prediction is now often run at proteome scale, where the goal is to move an entire worklist through the pipeline efficiently.
Elizabeth Goodman
10 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Every NVIDIA CUDA Toolkit release adds functionality and performance improvements that help developers get more from NVIDIA GPUs and the broader NVIDIA software…
Jonathan Bentz
14 min read
Includes Code
--
NVIDIA logo
NVIDIA
Intermediate
Open AI models are evolving faster than ever, but bringing them into native applications can still require model-specific conversion, preprocessing…
Tanya Lenz
6 min read
Includes Code
--
NVIDIA logo
NVIDIA
Intermediate
Alibaba released the model weights for Qwen3.8-Flash-Next as a preview of the upcoming Qwen4 architecture for developers to experiment with and evaluate.
Michelle Horton
3 min read
--
NVIDIA logo
NVIDIA
Advanced
For years, a Python developer who needed a GPU had two realistic choices: Learn NVIDIA CUDA C++ well enough to write an extension, set up a build toolchain…
Elizabeth Goodman
12 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Use AdaptGrow, a GPU-accelerated matrix factorization algorithm, to turn rolling correlation and tail-dependence matrices into hard clusters…
Elizabeth Goodman
12 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Recommender systems (RecSys) are one of the most ubiquitous machine learning problems in the consumer internet industry yet notoriously difficult to train and…
Elizabeth Goodman
11 min read
Includes Code
--
NVIDIA logo
NVIDIA
Intermediate
NVIDIA Holoscan is a platform for building real-time AI applications at the edge, from medical imaging to robotics. HoloHub is its companion repository: a…
Elizabeth Goodman
10 min read
Includes Code
--
NVIDIA logo
NVIDIA
Intermediate
Modern vision-language models (VLMs) can support tasks such as visual question answering, captioning, and image-text reasoning. In practice, however…
NVIDIA logo
NVIDIA
Intermediate
Atomistic simulation requires three things: knowledge of the science, compute-efficient implementation of simulations, and accessible interfaces to the…
Elizabeth Goodman
11 min read
Includes Code
--
NVIDIA logo
NVIDIA
Intermediate
Teams customize their models to hit their targets for latency, speed, memory, and compute. With the open NVIDIA Nemotron family of models…
Tanya Lenz
18 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Alibaba released the open weights for Qwen3.8-2.4T-A95B (Qwen3.8-Max), its largest open-weight model, bringing near-frontier capabilities to the open ecosystem.
Michelle Horton
4 min read
--
NVIDIA logo
NVIDIA
Advanced
NVIDIA nvmath-python is a library designed to bridge the gap between the Python scientific community and NVIDIA CUDA-X math libraries. It gives Python users…
Michelle Horton
14 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Frontier model pre-training has converged on mixture of experts (MoE), which is fundamentally changing what limits large-scale AI training. As compute per token…
Kirthi Devleker
7 min read
--
NVIDIA logo
NVIDIA
Intermediate
Developers building 3D, design, simulation, robotics, and industrial digital twin applications need ways to bring physical AI capabilities into the tools and…
Tanya Lenz
16 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Useful quantum computers will require fault tolerant logical operations. Researchers are actively exploring many different quantum error correction (QEC) codes…
Elizabeth Goodman
5 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
There are many ways to optimize code for GPUs. In this post, you’ll learn how kernel fusion can improve memory bandwidth and reduce kernel launch overhead…
Michelle Horton
10 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Biomolecular structure prediction and co-folding with models like OpenFold3 are now mainstream, large-scale workloads powering drug discovery and protein design.
Elizabeth Goodman
8 min read
Includes Code
--
NVIDIA logo
NVIDIA
Intermediate
NVIDIA Omniverse NuRec is a neural reconstruction pipeline for building high-fidelity 3D representations of real-world environments from multisensor data such…
Tanya Lenz
8 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Generative AI workloads are rapidly outgrowing the memory and compute budget of single GPUs. For inference developers building media generation pipelines…
Peter Kisfaludi
10 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
As AI systems move from single-turn interactions to coordinated multiagent workflows, low-latency inference becomes increasingly important.
Amr Elmeleegy
7 min read
--
NVIDIA logo
NVIDIA
Advanced
Foundation models are reshaping computational biology. Pretrained on massive corpora of protein or genomic sequences, models such as ESM2 (a protein language…
Bruno Alvisio
11 min read
Includes Code
--
NVIDIA logo
NVIDIA
Intermediate
As enterprise AI adoption scales, developers are increasingly forced to stitch together fragmented pipelines—separate models for text, vision…
Anu Srivastava
5 min read
Includes Code
--
NVIDIA logo
NVIDIA
Intermediate
This post is the third of a three-part series. See also Model Quantization: Concepts, Methods, and Why It Matters and Model Quantization: Post-Training…
Ruixiang Wang
9 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
AI agents are changing how you interact with your PC. Creators, developers, and AI enthusiasts are already using these agents extensively to assist with day-to…
Annamalai Chockalingam
8 min read
--
NVIDIA logo
NVIDIA
Advanced
Each wave of AI has created a new scaling law. Pretraining scaled intelligence through larger datasets, more parameters, and massively parallel GPU systems.
Praveen Menon
7 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
AI applications are moving beyond text generation to multimodal systems that can perceive, search, and reason across images, documents, video…
Anu Srivastava
4 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
In production inference deployments, demand fluctuates over time, requiring inference replicas to scale elastically. However, cold-starting inference workloads…
Schwinn Saereesitthipitak
14 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Large language models (LLMs) are revolutionizing the financial trading landscape by enabling sophisticated analysis of vast amounts of unstructured data to…
Dan Blanaru
10 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
NVIDIA CUDA 13.3 brings new capabilities and performance optimizations to developers across the CUDA ecosystem. The launch of NVIDIA CUDA Tile programming in…
Jonathan Bentz
13 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
The path from a trained AI model to production should be smooth, but rarely is. Many teams invest weeks fine-tuning models, only to discover that exporting to a…
Lovina Dmello
10 min read
Includes Code
--
NVIDIA logo
NVIDIA
Intermediate
This post is the second of a three-part series. See also Model Quantization: Concepts, Methods, and Why It Matters and Model Quantization: Turn FP8 Checkpoints…
Ruixiang Wang
8 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
For decades, computational biology has operated under a reductionist compromise. To fit complex biological systems into the limited memory of a single GPU…
Dejun Lin
8 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Federated learning (FL) is no longer a research curiosity—it’s a practical response to a hard constraint: the most valuable data is often the least movable.
NVIDIA logo
NVIDIA
Intermediate
In March 2026, three LLM agents generated over 600,000 lines of code, ran 850 experiments, and helped secure a first-place finish in a Kaggle playground…
NVIDIA logo
NVIDIA
Advanced
In a previous post, we introduced the Universal Sparse Tensor (UST), enabling developers to decouple a tensor’s sparsity from its memory layout for greater…
Aart J.C. Bik
10 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
Higher-order optimization algorithms such as Shampoo have been effectively applied in neural network training for at least a decade. These methods have achieved…
Hao Wu
9 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
The development of socially acceptable nuclear reactors requires that they are safe, clean, efficient, economical, and sustainable.
Mark Hobbs
11 min read
--
NVIDIA logo
NVIDIA
Advanced
For decades, computational chemistry has faced a tug-of-war between accuracy and speed. Ab initio methods like density functional theory (DFT) provide high…
Erica Tsai
13 min read
Includes Code
--
NVIDIA logo
NVIDIA
Advanced
NVIDIA Ising is the world’s first family of open AI models for building quantum processors, launching with two model domains: Ising Calibration and Ising…
NVIDIA logo
NVIDIA
Intermediate
Training LLMs requires periodic checkpoints. These full snapshots of model weights, optimizer states, and gradients are saved to storage so training can resume…
Wenqi Glantz
12 min read
Includes Code
--
NVIDIA logo
NVIDIA
Intermediate
Physical AI—AI systems that perceive, reason, and act in physically grounded simulated environments—is changing how teams design and validate robots and…
NVIDIA logo
NVIDIA
Intermediate
The Gemmaverse expands with the launch of the latest Gemma 4 multimodal and multilingual models, designed to scale across the full spectrum of deployments…
NVIDIA logo
NVIDIA
Advanced
NVIDIA Groq 3 LPX is a new rack-scale inference accelerator for the NVIDIA Vera Rubin platform, designed for the low-latency and large-context demands of…
NVIDIA logo
NVIDIA
Advanced
Computer-aided engineering (CAE) is shifting from human-driven workflows toward AI-driven ones, including physics foundation models that generalize across…