Agent skill · nvidia
tao-mine-nearest-neighbors
Run TAO Data Services TMM nearest-neighbor mining from embedding parquet files. Use when a workflow needs to mine source samples closest to target samples.
What it needs
About 3k tokens when loaded.
What this skill does
TAO Mine Nearest Neighbors Use this skill to run TAO Data Services TMM nearest-neighbor mining. The skill consumes embedding parquets and writes a mined source-sample parquet plus a mining summary. It does not compute embeddings; upstream steps must produce the source and target embedding parquets first. The container entrypoint is: TAO Data Services requires -e/--experimentspecfile. The tmm console script converts that YAML into Hydra --config-path and --config-name arguments internally. Inputs The user can provide either an existing nearest-neighbors YAML spec or the fields needed to generate one. Required spec fields: Field Meaning --- --- sourceparquet Absolute path to the candidate/source embeddings parquet. targetparquet Absolute path to the target/query embeddings parquet. outputparquet Absolute path where TAO Data Services should write mined source filepaths. Common optional fields: Field Default Meaning --- ---: --- topn 5 Number of nearest source samples to retrieve per target sample. knnmetric cosine One of cosine, euclidean, or manhattan. sourceembedcolumnname embedding Embedding column in sourceparquet. targetembedcolumnname embedding Embedding column in targetparquet. filterbylabel "false" String flag. When "true", TAO DS filters neighbors by matching label columns when both parquets provide labels. distancethreshold -1.0 Maximum distance to keep. Negative disables thresholding. Both input parquets must contain a filepath column and a list-like embedding column. If filterbylabel is "true", both parquets should also contain label. The default template is assets/defaultnearestneighbors.yaml. Quick Start Run from the tao-skill-bank repo root. Resolve the pinned TAO Data Services image from versions.yaml, verify the spec, mount the run root with identical host/container paths, and stream the Docker logs. Use at least one GPU. Choose GPUCOUNT from the hardware available to the host or platform that will run the container. …
How to use it
Reference it in AdaL, Claude Code, Cursor or any coding agent — nothing to install:
@skills nvidia/tao-mine-nearest-neighbors