跳到正文
cccskills

“pytorch”的搜索结果

236 个 · 按相关度排序

Guides PyTorch model training and fine-tuning, from choosing architectures to configuring optimizers and learning rates. Provides troubleshooting steps for unstable loss and memory shortages, plus methods for improving GPU throughput and comparing experiments.

中文简介正在准备中。当前显示英语简介。

  • Choose model architectures for the available data type and scale
  • Configure optimizers and learning-rate schedules
  • Investigate loss spikes, NaNs, and GPU memory shortages
Orchestra-Research/AI-Research-SKILLs1.4万2026年10月11日 更新

Generate C/C++ or CUDA code from an AI model (PyTorch, LiteRT) using MATLAB Coder or GPU Coder. Use when the user wants to integrate an AI model into an application with code generation as the end goal — generating MEX, CUDA MEX, static library, dynamic library, or executable — or using the model in Simulink for simulation and code generation. Covers PyTorch ExportedProgram (.pt2) via loadPyTorchExportedProgram and LiteRT (.tflite) via loadLiteRTModel (R2026a+). Keywords: PyTorch, torch, .pt2, ExportedProgram, loadPyTorchExportedProgram, invoke, codegen, MEX, CUDA, GPU, C, C++, deploy, AI model, deep learning model, LiteRT, TFLite, TensorFlow Lite, Simulink, slbuild, PyTorch ExportedProgram block, MATLAB Function block, dlosslib, loadLiteRTModel.

中文简介正在准备中。当前显示原文说明。

matlab/matlab-agentic-toolkit11812026年10月14日 更新

Implement and debug large-model training with PyTorch FSDP, which splits training state across devices. Consult documentation and code patterns for sharding choices, mixed precision, and checkpoints when building or reviewing distributed training code.

中文简介正在准备中。当前显示英语简介。

  • Implement distributed training for large models
  • Compare FSDP wrapping and sharding patterns
  • Review mixed-precision and checkpoint examples
NousResearch/hermes-agent25.5万2026年10月19日 更新

Integrate Flash Attention into Transformer models to speed up attention, the calculation that relates input elements, while reducing GPU memory use. Covers PyTorch integration, performance measurement, output comparison, and a separate H100 optimization path.

中文简介正在准备中。当前显示英语简介。

  • Optimize training with long input sequences
  • Reduce memory use during long-context inference
  • Measure speed and compare model outputs
NousResearch/hermes-agent25.5万2026年10月19日 更新

Set up model training with torchforge for reinforcement learning, where rewards guide improvement. Covers training configurations, custom reward and loss functions, and distributed execution with Monarch and TorchTitan, keeping algorithm experiments separate from infrastructure concerns.

中文简介正在准备中。当前显示英语简介。

  • Train mathematical reasoning models with GRPO
  • Experiment with custom reward and loss functions
  • Scale training across multiple GPUs or nodes
davila7/claude-code-templates3.3万2026年10月18日 更新

Investigate why a PyTorch model behaves a certain way by changing its internal values with pyvene. The skill guides experiments that swap or remove activations, train interventions, compare outputs, and save or share intervention configurations.

中文简介正在准备中。当前显示英语简介。

  • Locating model components that affect an output
  • Swapping activations to test behavioral hypotheses
  • Training interventions to investigate causal structure
davila7/claude-code-templates3.3万2026年10月18日 更新

Guides experiments that inspect and change internal model values using nnsight. Covers local PyTorch models and remote experiments through NDIF, including replacing intermediate values to test how particular layers affect a model’s predictions.

中文简介正在准备中。当前显示英语简介。

  • Inspecting layer outputs and attention patterns
  • Replacing internal values to compare predictions
  • Running large-model experiments through NDIF
davila7/claude-code-templates3.3万2026年10月18日 更新

ray-data

免费

Build data processing pipelines for machine learning, from preprocessing to batch predictions. Uses Ray Data to handle datasets larger than memory, distribute work across CPUs and GPUs, and feed data into training frameworks.

中文简介正在准备中。当前显示英语简介。

  • Preprocessing datasets larger than memory
  • Running batch predictions across multiple GPUs
  • Preparing image and tabular data for training
davila7/claude-code-templates3.3万2026年10月18日 更新

ray-data

免费

Build data processing pipelines for machine learning, from preprocessing to batch predictions. Uses Ray Data to handle datasets larger than memory, distribute work across CPUs and GPUs, and feed data into training frameworks.

中文简介正在准备中。当前显示英语简介。

  • Preprocessing datasets larger than memory
  • Running batch predictions across multiple GPUs
  • Preparing image and tabular data for training
Orchestra-Research/AI-Research-SKILLs1.4万2026年10月11日 更新

Guides experiments that inspect and modify internal values in PyTorch models to study their effects on predictions. Covers local analysis and remote experiments through NDIF for models too large for local GPUs.

中文简介正在准备中。当前显示英语简介。

  • Inspect internal values and attention patterns across layers
  • Replace internal values to compare changes in predictions
  • Run large-model experiments without local GPUs
Orchestra-Research/AI-Research-SKILLs1.4万2026年10月11日 更新

Guides reinforcement learning, where models learn from rewards, using torchforge and PyTorch. Covers reward design, custom loss functions, and distributed training so researchers can experiment with algorithms while keeping infrastructure concerns separate.

中文简介正在准备中。当前显示英语简介。

  • Train mathematical reasoning models with GRPO
  • Experiment with custom loss functions
  • Scale training across multiple GPUs or nodes
Orchestra-Research/AI-Research-SKILLs1.4万2026年10月11日 更新

Guides experiments that change internal values in PyTorch models to test what causes their behavior. Uses pyvene to swap or remove activations, locate influential components, train interventions, and save experiments for reuse.

中文简介正在准备中。当前显示英语简介。

  • Locating model components that affect an answer
  • Comparing outputs after swapping internal activations
  • Training interventions to investigate causal structure
Orchestra-Research/AI-Research-SKILLs1.4万2026年10月11日 更新

Convert existing plain or manual PyTorch training code into an NVFLARE federated job using Client API model exchange, local validation, and job export; use when the user names plain PyTorch or preliminary source inspection identifies one plain-PyTorch owner, and not for Lightning, other frameworks, deployment, POC/production lifecycle, or experiment workflows.

中文简介正在准备中。当前显示原文说明。

NVIDIA/skills36432026年10月18日 更新

Set up large language model pretraining with TorchTitan and PyTorch. Configure training across multiple GPUs, combine four ways of splitting the workload, enable Float8 calculations on H100 hardware, and manage checkpoints for saving and resuming training.

中文简介正在准备中。当前显示英语简介。

  • Pretraining Llama 3.1 8B on a single node
  • Launching training across multiple nodes with SLURM
  • Configuring Float8 training for H100 GPUs
NousResearch/hermes-agent25.5万2026年10月19日 更新

Set up machine learning training and inference on NVIDIA DGX Spark. Get guidance on choosing containers and packages, resolving CUDA compatibility errors, and checking that PyTorch can access the GPU before starting a workload.

中文简介正在准备中。当前显示英语简介。

  • Setting up a new DGX Spark for training or inference
  • Diagnosing CUDA compatibility and import errors
  • Choosing between containers and direct pip installation
wshobson/agents4.1万2026年10月19日 更新

Evaluate a machine learning GPU cluster during a cloud trial or acceptance test. Measure GPU computing performance, communication within and between machines, and local and shared storage, then compile the results into a dated Markdown report.

中文简介正在准备中。当前显示英语简介。

  • Evaluating trial GPU machines from a cloud provider
  • Running acceptance tests for a GPU cluster
  • Measuring GPU performance under load from neighboring GPUs
stas00/ml-engineering2万2026年10月17日 更新

Convert existing PyTorch Lightning training code into an NVFLARE federated job using the Lightning Client API patch, local validation, and job export; use only when the request names federated/NVFLARE conversion or asks multiple sites to train collaboratively while keeping each site's data local, and either names PyTorch Lightning or preliminary source inspection identifies one Lightning owner; do not use for non-federated Lightning work such as DDP, profiling, inference serving, or training-loop changes, nor for plain PyTorch, TensorFlow/Keras, other frameworks, deployment, POC/production lifecycle, or experiment workflows.

中文简介正在准备中。当前显示原文说明。

NVIDIA/skills36432026年10月18日 更新

Deploy AI models to embedded hardware using MathWorks tools (MATLAB, Simulink, Embedded Coder). Covers two workflow patterns: (1) MathWorks-native or imported models rebuilt as dlnetwork for lean hardware, (2) direct C/C++ code generation from PyTorch and LiteRT models. Both patterns support all targets (Cortex-M/A/R, x86, GPU). Trigger when: user wants to deploy AI to embedded targets; generate C/CUDA from neural networks; compress AI models for MCU; integrate AI in Simulink for system-level simulation; import PyTorch/ONNX/TensorFlow models for embedded deployment; optimize AI for resource-constrained hardware; or use loadPyTorchExportedProgram, loadLiteRTModel, importNetworkFromPyTorch, importNetworkFromONNX, importNetworkFromTensorFlow, importNetworkFromKeras, dlquantizer, exportNetworkToSimulink, or Embedded Coder with AI models.

中文简介正在准备中。当前显示原文说明。

matlab/matlab-agentic-toolkit11812026年10月14日 更新

Import PyTorch, ONNX, or Keras 3 / TensorFlow 2.16+ deep learning models into MATLAB as dlnetwork objects. Use when importing .pt2 exported programs, traced .pt files, .onnx models, or Keras 3 models via matlabsaver. Covers importNetworkFromPyTorch, importNetworkFromONNX, importNetworkFromKeras, importNetworkFromTensorFlow, torch.export.export, PyTorchInputSizes, InputDataFormats, matlabsaver, tf_keras downgrade, numeric validation against PyTorch or ONNX Runtime, and placeholder/custom layer implementation. Applies when user mentions any of these functions, file formats, or encounters import errors, unsupported operator warnings, 0 learnables, or uninitialized networks.

中文简介正在准备中。当前显示原文说明。

matlab/matlab-agentic-toolkit11812026年10月14日 更新

Deploy AI models to embedded hardware using MathWorks tools (MATLAB, Simulink, Embedded Coder). Covers two workflow patterns: (1) MathWorks-native or 3P-imported models rebuilt as dlnetwork for lean hardware (Cortex-M, DSP), (2) direct C/C++ code generation from PyTorch and LiteRT models for high-performance hardware (Cortex-A, x86, GPU). Trigger when: user wants to deploy AI to embedded targets; generate C/CUDA from neural networks; compress AI models for MCU/DSP; integrate AI in Simulink for system-level simulation; import PyTorch/ONNX/TensorFlow models for embedded deployment; optimize AI for resource-constrained hardware; or use loadPyTorchExportedProgram, importNetworkFromPyTorch, dlquantizer, exportNetworkToSimulink, or Embedded Coder with AI models.

中文简介正在准备中。当前显示原文说明。

matlab/agent-skills-playground1882026年10月15日 更新

面向 Ascend PyTorch Profiler / msprof DB(如 ascend_pytorch_profiler*.db、msprof_*.db)的 SQL 分析技能。将自然语言问题(算子耗时、通信、下发、调度、schema/table 查询)转为安全可执行 SQL,并按需从官方文档提取表结构详情。

中文简介正在准备中。当前显示原文说明。

kali20gakki/msAgent322026年10月17日 更新

Use Hugging Face Transformers to load pretrained models, generate text, and classify text, images, or audio. Prepare model inputs, control generation settings, and fine-tune models on custom datasets with the Trainer API.

中文简介正在准备中。当前显示英语简介。

  • Prototype text, image, or audio classification
  • Configure text generation and conversational responses
  • Fine-tune a model on a custom dataset
K-Dense-AI/scientific-agent-skills5.3万2026年10月16日 更新

Use W&B to record machine learning experiments, visualize training, and compare results. The skill also covers automated searches for training settings and version tracking for datasets and models, helping teams share experiments and trace their inputs.

中文简介正在准备中。当前显示英语简介。

  • Logging training metrics and viewing progress in real time
  • Comparing experiments with different training settings
  • Searching automatically for better training settings
davila7/claude-code-templates3.3万2026年10月18日 更新

Build visual AI systems that detect objects or divide images into labeled regions. This skill guides dataset preparation, model selection, training, evaluation, inference optimization, and deployment using frameworks such as PyTorch and Ultralytics.

中文简介正在准备中。当前显示英语简介。

  • Build an object detection pipeline
  • Clean images and convert annotation formats
  • Optimize a trained model for deployment
alirezarezvani/claude-skills2.8万2026年8月30日 更新