skills.homes logoskills.homescapability registry
home/categories/machine-learning/deepmodeling-deepmd-kit-github-skills-debug-gradient-flow-skill-md
cd ..
machine-learningdata-ai

debug-gradient-flow

Diagnose gradient flow issues in training, especially for compiled models (torch.compile/make_fx). Systematically isolates which loss components (energy, force, virial) contribute gradients to which parameters, and identifies where the gradient chain breaks.

View Sourcemachine-learning
deepmodeling
maintainer
deepmodeling
Updated 3/3/2026
Stars
1909
Forks
607
quick start

Installation and usage

Diagnose gradient flow issues in training, especially for compiled models (torch.compile/make_fx). Systematically isolates which loss components (energy, force, virial) contribute gradients to which parameters, and identifies where the gradient chain breaks.

Installation
$ install --globalskills.sh
Usage

Once installed, you can use this skill by running the following command in your terminal:

skills use debug-gradient-flow