← all repositories

meta-pytorch/captum

A PyTorch library for model interpretability providing implementations of integrated gradients, saliency maps, smoothgrad, and other feature attribution methods.

5.6k stars Python ML FrameworksLLMOps · Eval
captum
Velocity · 7d
+2.3
★ / day
Trend
steady
star history

Captum is Meta’s official model interpretability library for PyTorch, implementing state-of-the-art algorithms for understanding how neural network models make decisions. It provides general-purpose implementations of attribution methods including integrated gradients, saliency maps, SmoothGrad, and VarGrad. The library integrates with domain-specific PyTorch libraries like torchvision and torchtext, enabling interpretability analysis across computer vision, NLP, and other model architectures.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.