Machine Learning
The best skills for data-related work, like analysis, visualization, SQL, pipelines and machine learning.
tao-convert-dataset-formatNVIDIA/skillsConvert TAO DAFT datasets between supported formats with the tao-daft CLI★3.8K
tao-convert-dataset-formatConvert TAO DAFT datasets between supported formats with the tao-daft CLI
tao-data-ioNVIDIA/skillsStage datasets, map credentials and route outputs for TAO training jobs★3.8K
tao-data-ioStage datasets, map credentials and route outputs for TAO training jobs
tao-finetune-anomalygennextNVIDIA/skillsValidate an AnomalyGenNext dataset and prepare its texture fine-tuning recipe★3.8K
tao-finetune-anomalygennextValidate an AnomalyGenNext dataset and prepare its texture fine-tuning recipe
tao-finetune-clipNVIDIA/skillsFine-tune CLIP for retrieval and zero-shot tasks and export to ONNX or TensorRT★3.8K
tao-finetune-clipFine-tune CLIP for retrieval and zero-shot tasks and export to ONNX or TensorRT
tao-finetune-cosmos-embedNVIDIA/skillsFine-tune Cosmos-Embed1 for text-to-video search and semantic deduplication★3.8K
tao-finetune-cosmos-embedFine-tune Cosmos-Embed1 for text-to-video search and semantic deduplication
tao-finetune-cosmos-reasonNVIDIA/skillsFine-tune Cosmos3 models on Cosmos Framework or Cosmos-RL with validated inputs★3.8K
tao-finetune-cosmos-reasonFine-tune Cosmos3 models on Cosmos Framework or Cosmos-RL with validated inputs
tao-finetune-huggingface-modelNVIDIA/skillsFine-tune any Hugging Face vision, VLM or LLM model on local NVIDIA GPUs★3.8K
tao-finetune-huggingface-modelFine-tune any Hugging Face vision, VLM or LLM model on local NVIDIA GPUs
tao-finetune-kumo-anomalyNVIDIA/skillsRun and fine-tune Kumo-Anomaly diffusion detection on multivariate time series★3.8K
tao-finetune-kumo-anomalyRun and fine-tune Kumo-Anomaly diffusion detection on multivariate time series
tao-finetune-kumo-forecastNVIDIA/skillsRun and fine-tune Kumo-Forecast with DARR retrieval on multivariate time series★3.8K
tao-finetune-kumo-forecastRun and fine-tune Kumo-Forecast with DARR retrieval on multivariate time series
tao-finetune-video-clipNVIDIA/skillsFine-tune InternVideo2-CLIP for video-text retrieval and export it to TensorRT★3.8K
tao-finetune-video-clipFine-tune InternVideo2-CLIP for video-text retrieval and export it to TensorRT
tao-generate-anomaliesNVIDIA/skillsFine-tune AnomalyGen and generate synthetic anomaly images with quality scoring★3.8K
tao-generate-anomaliesFine-tune AnomalyGen and generate synthetic anomaly images with quality scoring
tao-generate-image-embeddingsNVIDIA/skillsTurn a parquet list of image paths into CLIP or SigLIP embedding files★3.8K
tao-generate-image-embeddingsTurn a parquet list of image paths into CLIP or SigLIP embedding files
tao-generate-image-groundingNVIDIA/skillsGround captions to pixel bounding boxes with a two-step VLM labeling pipeline★3.8K
tao-generate-image-groundingGround captions to pixel bounding boxes with a two-step VLM labeling pipeline
tao-generate-od-defectsNVIDIA/skillsGenerate synthetic defects with AnomalyGenNext and publish COCO labels★3.8K
tao-generate-od-defectsGenerate synthetic defects with AnomalyGenNext and publish COCO labels
tao-generate-referring-expressionsNVIDIA/skillsGenerate referring-expression annotations from images and KITTI boxes with a VLM★3.8K
tao-generate-referring-expressionsGenerate referring-expression annotations from images and KITTI boxes with a VLM
tao-generate-video-reasoning-annotationsNVIDIA/skillsTurn raw videos into chain-of-thought training data, captions and QA pairs★3.8K
tao-generate-video-reasoning-annotationsTurn raw videos into chain-of-thought training data, captions and QA pairs
tao-launch-workflowNVIDIA/skillsEnforce a pre-launch check and run contract for every TAO training or export job★3.8K
tao-launch-workflowEnforce a pre-launch check and run contract for every TAO training or export job
tao-list-capabilitiesNVIDIA/skillsList the TAO models, workflows and AutoML options available in the skill bank★3.8K
tao-list-capabilitiesList the TAO models, workflows and AutoML options available in the skill bank
tao-mine-aoi-imagesNVIDIA/skillsEmbed AOI gap targets and mine nearest source images for augmentation★3.8K
tao-mine-aoi-imagesEmbed AOI gap targets and mine nearest source images for augmentation
tao-mine-nearest-neighborsNVIDIA/skillsMine source samples nearest to your targets from embedding parquet files★3.8K
tao-mine-nearest-neighborsMine source samples nearest to your targets from embedding parquet files
tao-mine-od-imagesNVIDIA/skillsMine unique source images for object detection from embedding parquet files★3.8K
tao-mine-od-imagesMine unique source images for object detection from embedding parquet files
tao-port-huggingface-modelNVIDIA/skillsWire a Hugging Face vision model into TAO config, trainer and TensorRT pipeline★3.8K
tao-port-huggingface-modelWire a Hugging Face vision model into TAO config, trainer and TensorRT pipeline
tao-prepare-anomalygennext-inputsNVIDIA/skillsPrepare false negatives, source masks and embeddings as AnomalyGenNext inputs★3.8K
tao-prepare-anomalygennext-inputsPrepare false negatives, source masks and embeddings as AnomalyGenNext inputs
tao-route-visual-changenet-samplesNVIDIA/skillsSplit weak Visual ChangeNet samples into per-augmentation subsets by eligibility★3.8K
tao-route-visual-changenet-samplesSplit weak Visual ChangeNet samples into per-augmentation subsets by eligibility
tao-run-automlNVIDIA/skillsRun TAO AutoML hyperparameter search with Bayesian, ASHA, BOHB or LLM methods★3.8K
tao-run-automlRun TAO AutoML hyperparameter search with Bayesian, ASHA, BOHB or LLM methods
tao-run-automl-deft-pipelineNVIDIA/skillsRun the three-phase AOI pipeline: AutoML baseline, DEFT loop and refinement★3.8K
tao-run-automl-deft-pipelineRun the three-phase AOI pipeline: AutoML baseline, DEFT loop and refinement
tao-run-deft-aoiNVIDIA/skillsRun the full DEFT loop to improve PCB inspection models with synthetic defects★3.8K
tao-run-deft-aoiRun the full DEFT loop to improve PCB inspection models with synthetic defects
tao-run-deft-aoi-cosmos3NVIDIA/skillsRun the disk-backed DEFT AOI improvement loop for Cosmos Reason 3 models★3.8K
tao-run-deft-aoi-cosmos3Run the disk-backed DEFT AOI improvement loop for Cosmos Reason 3 models
tao-run-deft-cr-its-miningNVIDIA/skillsRun a mining-based DEFT loop for traffic-camera Cosmos Reason yes or no questions★3.8K
tao-run-deft-cr-its-miningRun a mining-based DEFT loop for traffic-camera Cosmos Reason yes or no questions
tao-run-deft-object-detectionNVIDIA/skillsRun the DEFT augmentation loop to retrain Grounding DINO object detectors★3.8K
tao-run-deft-object-detectionRun the DEFT augmentation loop to retrain Grounding DINO object detectors
tao-run-deft-od-aoiNVIDIA/skillsRun a binary inspection DEFT loop with RT-DETR and select models by AP50★3.8K
tao-run-deft-od-aoiRun a binary inspection DEFT loop with RT-DETR and select models by AP50
tao-run-deft-pasNVIDIA/skillsIteratively improve CLIP or SigLIP retrieval on attribute-labelled image data★3.8K
tao-run-deft-pasIteratively improve CLIP or SigLIP retrieval on attribute-labelled image data
tao-run-dinov3-ssl-deftNVIDIA/skillsSelect unlabeled images for DINOv3 retraining in an iterative active-learning loop★3.8K
tao-run-dinov3-ssl-deftSelect unlabeled images for DINOv3 retraining in an iterative active-learning loop
tao-run-inference-serviceNVIDIA/skillsStart, query and stop a TAO inference microservice in a container★3.8K
tao-run-inference-serviceStart, query and stop a TAO inference microservice in a container
tao-run-on-virtualenvNVIDIA/skillsRun a Python training or eval script in an existing virtualenv, no containers★3.8K
tao-run-on-virtualenvRun a Python training or eval script in an existing virtualenv, no containers
tao-train-action-recognitionNVIDIA/skillsTrain, evaluate and export TAO action recognition models on video clips★3.8K
tao-train-action-recognitionTrain, evaluate and export TAO action recognition models on video clips
tao-train-bevfusionNVIDIA/skillsTrain BEVFusion models that fuse LiDAR and camera data for 3D object detection★3.8K
tao-train-bevfusionTrain BEVFusion models that fuse LiDAR and camera data for 3D object detection
tao-train-centerposeNVIDIA/skillsTrain CenterPose models for 6-DoF object pose and keypoint estimation★3.8K
tao-train-centerposeTrain CenterPose models for 6-DoF object pose and keypoint estimation
tao-train-codetrNVIDIA/skillsTrain Co-DETR object detection models with hybrid assignment in TAO★3.8K
tao-train-codetrTrain Co-DETR object detection models with hybrid assignment in TAO
tao-train-deformable-detrNVIDIA/skillsTrain, quantize and export Deformable DETR models for 2D object detection★3.8K
tao-train-deformable-detrTrain, quantize and export Deformable DETR models for 2D object detection
tao-train-depth-anything-v2NVIDIA/skillsTrain monocular depth estimation models based on Depth Anything v2★3.8K
tao-train-depth-anything-v2Train monocular depth estimation models based on Depth Anything v2
tao-train-dinoNVIDIA/skillsTrain DINO object detectors with distillation and quantization in TAO★3.8K
tao-train-dinoTrain DINO object detectors with distillation and quantization in TAO
tao-train-dinov3NVIDIA/skillsDomain-adapt DINOv3 vision backbones on unlabeled images with self-supervision★3.8K
tao-train-dinov3Domain-adapt DINOv3 vision backbones on unlabeled images with self-supervision
tao-train-fast-foundation-stereoNVIDIA/skillsTrain and run FastFoundationStereo for real-time stereo depth from image pairs★3.8K
tao-train-fast-foundation-stereoTrain and run FastFoundationStereo for real-time stereo depth from image pairs
tao-train-foundation-stereoNVIDIA/skillsTrain FoundationStereo models that predict disparity from stereo image pairs★3.8K
tao-train-foundation-stereoTrain FoundationStereo models that predict disparity from stereo image pairs
tao-train-grounding-dinoNVIDIA/skillsTrain Grounding DINO for open-set object detection driven by text prompts★3.8K
tao-train-grounding-dinoTrain Grounding DINO for open-set object detection driven by text prompts
tao-train-image-classificationNVIDIA/skillsTrain, distill and quantize PyTorch image classifiers with many backbones★3.8K
tao-train-image-classificationTrain, distill and quantize PyTorch image classifiers with many backbones
tao-train-mask-auto-encoderNVIDIA/skillsPretrain and fine-tune Masked Auto-Encoder backbones for vision tasks★3.8K
tao-train-mask-auto-encoderPretrain and fine-tune Masked Auto-Encoder backbones for vision tasks
tao-train-mask-auto-labelNVIDIA/skillsTrain Mask Auto-Label models that build segmentation masks from point or box hints★3.8K
tao-train-mask-auto-labelTrain Mask Auto-Label models that build segmentation masks from point or box hints
tao-train-mask-grounding-dinoNVIDIA/skillsTrain Mask Grounding DINO for text-prompted instance segmentation★3.8K
tao-train-mask-grounding-dinoTrain Mask Grounding DINO for text-prompted instance segmentation