GPU-optimized AI, Machine Learning, & HPC Software

SearchSearch thousands of GPU-optimized Containers, pretrained Models, SDKs, and Helm charts—ready to accelerate AI, digital twins, and HPC from cloud to edge.

Displaying 36 results

NVIDIA

PeopleSemSegnet

Semantic segmentation of persons in an image.

Model

NVIDIA

TAO Toolkit API Helm

Helm Chart to deploy TAO Toolkit APIs

Helm Chart

NVIDIA

TAO Pretrained Object Detection

Pretrained weights to facilitate transfer learning using TAO Toolkit.

Model

Pre-trained SegFormer ImageNet weights

Pre-trained SegFormer weights trained on ImageNet to facilitate transfer learning using TAO Toolkit.

Model

—

ReIdentificationNet

Re-Identification network to generate embeddings for identifying persons in different scenes.

Model

NVIDIA

Optical Character Detection

Network to detect characters in an image.

Model

—

Pre-trained DINO NvImageNet weights

Pre-trained DINO weights trained on NvImageNet to facilitate transfer learning using TAO Toolkit.

Model

—

Pre-trained FAN based NVImageNet Classification weights

Pre-trained FAN weights trained on NVImageNet to facilitate transfer learning using TAO Toolkit.

Model

NVIDIA

PointPillarNet

Model to detect one or more objects from a LIDAR point cloud file and return 3D bounding boxes.

Model

—

Optical Inspection

Model to detect defects in soldered components on a Printed Circuit Board.

Model

—

Pre-trained FAN based ImageNet Classification weights

Pre-trained FAN weights trained on ImageNet to facilitate transfer learning using TAO Toolkit.

Model

—

Pre-trained Deformable DETR NvImageNet weights

Pre-trained deformable_detr weights trained on NvImageNet to facilitate transfer learning using TAO Toolkit.

Model

NVIDIA

BodyPose3DNet

3D human pose estimation network to predict 34 keypoints in 3D of a person in an image.

Model

NVIDIA

Pose Classification

Pose classification network to classify poses of people from their skeletons.

Model

NVIDIA

Optical Character Recognition

Model to recognise characters from a preceding OCDNet model.

Model

NVIDIA

CitySemSegFormer

Semantic segmentation of persons in an image.

Model

—

Retail Object Recognition

Embedding generator model to recognize objects on a checkout counter.

Model

—

DINO

Model trained on COCO2017 to detect and classify objects.

Model

—

Retail Object Detection

DINO (DETR with Improved DeNoising Anchor Boxes) based object detection network to detect retail objects on a checkout counter.

Model

NVIDIA

Mask Auto Label

Pretrained model to generate semantic segmentation labels.

Model

—

Pre-trained GCViT ImageNet Classification weights

Pre-trained GCViT weights trained on ImageNet to facilitate transfer learning using TAO Toolkit.

Model

—

Pre-trained EfficientDet NvImageNet backbones

Pre-trained EfficientNet backbone weights trained on NvImageNet to facilitate transfer learning using TAO Toolkit.

Model

—

Pre-trained DINO ImageNet weights

Pre-trained DINO weights trained on ImageNet to facilitate transfer learning using TAO Toolkit.

Model

NVIDIA

CenterPose

3 pose detection model for retail objects.

Model

1-24 of 36 items