viz.renderers.GradientFlowRenderer

class GradientFlowRenderer

GPU-accelerated gradient flow visualization.

Examples

const renderer = new GradientFlowRenderer();

// Collect gradient stats after backward pass
const stats = model.layers.map(layer => ({
  name: layer.name,
  mean: layer.weight.grad.abs().mean().item(),
  std: layer.weight.grad.std().item(),
}));

renderer.render(stats, { title: 'Gradient Flow' });