Skip to main content

vantage_sdk.workbench.nvidia_catalogs

SDK helpers for the NVIDIA AI stack discovery catalogs.

Classes​

BlueprintSDK()​

SDK for the curated NVIDIA AI Blueprint catalog.

Methods​

  • get(self, ctx, *, cluster_name: 'str', blueprint_id: 'str') -> 'ClusterServiceResponse': Get detail for one curated blueprint.
  • list(self, ctx, *, cluster_name: 'str', status: 'str | None' = None) -> 'ClusterServiceResponse': List curated blueprints, optionally filtered by availability.

NemoCatalogSDK()​

SDK for the read-only NeMo microservices catalog proxy.

Methods​

  • customization_configs(self, ctx, *, cluster_name: 'str') -> 'ClusterServiceResponse': List fine-tuning recipes served by the NeMo Customizer (model x technique).
  • customization_targets(self, ctx, *, cluster_name: 'str') -> 'ClusterServiceResponse': List fine-tunable base models registered with the NeMo Customizer.
  • evaluation_configs(self, ctx, *, cluster_name: 'str') -> 'ClusterServiceResponse': List evaluation configs registered with the NeMo Evaluator.

NgcContainerSDK()​

SDK for the curated NGC framework container catalog.

Methods​

  • get(self, ctx, *, cluster_name: 'str', name: 'str') -> 'ClusterServiceResponse': Get detail for one curated NGC container.
  • list(self, ctx, *, cluster_name: 'str', purpose: 'str | None' = None) -> 'ClusterServiceResponse': List curated NGC containers, optionally filtered by purpose.