Results for “pytorch”
15 skills
nvidia tao-train-image-classification
Train, evaluate, distill, quantize, export, and run inference for PyTorch-based TAO image classification models with support for multiple backbones.
2.2k · bundle
nvidia tao-finetune-huggingface-model
Fine-tune HuggingFace CV, VLM, or LLM models on local NVIDIA GPUs using an NGC PyTorch container, with support for full or LoRA training, dataset handling, and optional model push to the Hub.
2.2k · bundle
nvidia nemo-mbridge-multi-node-slurm
Convert single-node PyTorch distributed scripts into multi-node Slurm sbatch jobs and debug common multi-node failures, covering srun-native and torch.distributed approaches, container setup, NCCL timeouts, and interactive allocation.
2.2k · bundle
nvidia tao-train-segformer
Trains, evaluates, exports, quantizes, and runs inference for SegFormer semantic segmentation models using NVIDIA TAO.
2.2k · bundle
nvidia tilegym-adding-cutile-kernel
Add a new cuTile GPU kernel operator to TileGym, covering dispatch registration, backend implementation, exports, tests, and benchmarks.
2.2k · bundle
nvidia earth2studio-create-prognostic
Create Earth2Studio prognostic model wrappers that time-step weather forecasts forward, with triple-inheritance classes, tests, and documentation.
2.2k · bundle
More results
nvidia tao-train-pose-classification
Train, evaluate, export, and run inference for pose classification models using ST-GCN on skeleton keypoint sequences.
2.2k · bundle
nvidia nemo-mbridge-perf-megatron-fsdp
Enables Megatron Fully Sharded Data Parallel in Megatron-Bridge with configuration overrides, code anchors, pitfalls, and verification steps.
2.2k · bundle
nvidia tao-train-pointpillars
Train, evaluate, export, prune, and run inference for PointPillars 3D object detection models from LiDAR point clouds using NVIDIA TAO.
2.2k · bundle
nvidia dali-dynamic-mode
Write, review, and migrate code using NVIDIA DALI's imperative dynamic-mode API for efficient data loading and preprocessing.
2.2k · bundle
nvidia earth2studio-create-diagnostic
Create Earth2Studio diagnostic model wrappers for single-step data transformations, including simple derived diagnostics, packaged AutoModel diagnostics, and generative or diffusion diagnostics.
2.2k · bundle
nvidia nv-reason-cxr
Runs chest X-ray reasoning smoke tests using the NV-Reason-CXR-3B model via local inference or a public Hugging Face Space API.
2.2k · bundle
nvidia nemo-mbridge-mlm-bridge-training
Run Megatron-LM (MLM) and Megatron Bridge training with mock or real data, covering correlation testing, available recipes, and multi-GPU examples.
2.2k · bundle
nvidia tao-train-mask-auto-encoder
Train, evaluate, export, and run inference for Masked Auto-Encoder (MAE) models for self-supervised pretraining and fine-tuning of visual representations.
2.2k · bundle
nvidia nemo-mbridge-perf-memory-tuning
Reduces peak GPU memory in Megatron Bridge training by applying expandable segments, parallelism resizing, activation recompute, and CPU offloading constraints.
2.2k · bundle