PyTorch

Feb 13, 2025
Simplify System Memory Management with the Latest NVIDIA GH200 NVL2 Enterprise RA
NVIDIA Enterprise Reference Architectures (Enterprise RAs) can reduce the time and cost of deploying AI infrastructure solutions. They provide a streamlined...
8 MIN READ

Jan 30, 2025
New AI SDKs and Tools Released for NVIDIA Blackwell GeForce RTX 50 Series GPUs
NVIDIA recently announced a new generation of PC GPUs—the GeForce RTX 50 Series—alongside new AI-powered SDKs and tools for developers. Powered by the...
6 MIN READ

Sep 19, 2024
Just Released: Torch-TensorRT v2.4.0
Includes C++ runtime support in Windows Support, Enhanced Dynamic Shape support in Converters, PyTorch 2.4, CUDA 12.4, TensorRT 10.1, Python 3.12.
1 MIN READ

Sep 13, 2024
Improved Data Loading with Threads
Data loading is a critical aspect of deep learning workflows, whether you're focused on training or inference. However, it often presents a paradox: the need...
8 MIN READ

Jun 20, 2024
AI Brain Implant Restores Bilingual Communication for Stroke Survivor
Scientists have enabled a stroke survivor, who is unable to speak, to communicate in both Spanish and English by training a neuroprosthesis implant to decode...
3 MIN READ

Apr 26, 2024
Perception Model Training for Autonomous Vehicles with Tensor Parallelism
Due to the adoption of multicamera inputs and deep convolutional backbone networks, the GPU memory footprint for training autonomous driving perception models...
10 MIN READ

Jan 17, 2024
Release: PyTorch Geometric Container for GNNs on NGC
The NVIDIA PyG container, now generally available, packages PyTorch Geometric with accelerations for GNN models, dataloading, and pre-processing using...
1 MIN READ

Dec 08, 2023
Available Now: NVIDIA AI Accelerated DGL and PyG Containers for GNNs
From credit card transactions, social networks, and recommendation systems to transportation networks and protein-protein interactions in biology, graphs are...
9 MIN READ

Nov 20, 2023
Transforming Industrial Defect Detection with NVIDIA TAO and Vision AI Models
Efficiency is paramount in industrial manufacturing, where even minor gains can have significant financial implications. According to the American Society of...
11 MIN READ

Nov 08, 2023
New Workshop: Rapid Application Development Using Large Language Models
Interested in developing LLM-based applications? Get started with this exploration of the open-source ecosystem.
1 MIN READ

Oct 12, 2023
Workshop: Model Parallelism: Building and Deploying Large Neural Networks
Learn how to train the largest neural networks and deploy them to production.
1 MIN READ

Oct 04, 2023
Analyzing the Security of Machine Learning Research Code
The NVIDIA AI Red Team is focused on scaling secure development practices across the data, science, and AI ecosystems. We participate in open-source security...
12 MIN READ

Aug 09, 2023
Just Released: NVIDIA PhysicsNeMo 23.08
NVIDIA PhysicsNeMo is now part of the NVIDIA AI Enterprise suite, supporting PyTorch 2.0, CUDA 12, and new samples.
1 MIN READ

Jul 18, 2023
Research Unveils Breakthrough Deep Learning Tool for Understanding Neural Activity and Movement Control
A primary goal in the field of neuroscience is understanding how the brain controls movement. By improving pose estimation, neurobiologists can more precisely...
8 MIN READ

Jun 06, 2023
Develop Physics-Informed Machine Learning Models with Graph Neural Networks
NVIDIA PhysicsNeMo is a framework for building, training, and fine-tuning deep learning models for physical systems, otherwise known as physics-informed machine...
6 MIN READ

May 16, 2023
Sparsity in INT8: Training Workflow and Best Practices for NVIDIA TensorRT Acceleration
The training stage of deep learning (DL) models consists of learning numerous dense floating-point weight matrices, which results in a massive amount of...
12 MIN READ