Skip to main content

Search

Search thousands of GPU-optimized Containers, pretrained Models, SDKs, and Helm charts—ready to accelerate AI, digital twins, and HPC from cloud to edge.
Accelerate custom generative AI app deployment using pre-built containers with optimized AI models.

Displaying 8 results

PyG
NVIDIA
PyG (PyTorch Geometric) is a library built upon PyTorch to easily write and train Graph Neural Networks (GNNs) for a wide range of applications related to structured data.
Container
PyG Production Branch October 2024 (PB 24h2) offers a 9-month lifecycle for API stability, with monthly patches for high and critical software vulnerabilities.
Container
Atomistic Molecular Dynamics NIM with machine learning interatomic potentials.
Container
Atomistic Geometry Optimization/Relaxation NIM with machine learning interatomic potentials.
Container
This contains all the supplemental data for PhysicsNeMo Sym examples. This includes validation data, training data, large CSV/STLs, etc. For the python scripts, please refer https://github.com/nvidia/physicsnemo-sym
Resource
Stokes flow dataset with parameterized domain
Resource
This contains all the supplemental data for PhysicsNeMo examples. This includes files apart from the actual dataset required for training. For the python scripts, please refer https://github.com/nvidia/physicsnemo
Resource
This collection contains performance-optimized Deep Learning frameworks.
Collection
1-8 of 8 items

Modal Content

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.