captum
by pytorch
Model interpretability and understanding for PyTorch
AI summary
Model explainer
Provides tools and algorithms to understand how machine learning models make predictions
- stars
- 5K
- forks
- 503
- watching
- 278
- awesome lists
- 4
Similar projects
Found by comparing what the projects do, not just their names.
Optimization toolkit
A PyTorch-based library for Bayesian optimization, providing a modular interface for composing and optimizing probabilistic models.
PyTorch examples
A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning.
Image captioning framework
A Python-based framework for training and testing image captioning models using PyTorch.
Knowledge analyzer
Analyzing knowledge development and evolution in large language models during training
PyTorch C++ implementation
A C++ implementation of PyTorch tutorials
C++ PyTorch inference toolkit
A repository providing tools and examples to serve PyTorch models as C++ inference applications.
Deep Learning Tutorials
A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage.
Model deployment notes
A collection of notes and references on deploying deep learning models in production environments
Point Cloud Model
This is an implementation of the PointNet algorithm in PyTorch for 3D point cloud classification and segmentation tasks.
GPU tensor library
A Python library providing tensors and dynamic neural networks with strong GPU acceleration
Language Model
Trains a large-scale PyTorch language model on the 1-Billion Word dataset
BERT model
An implementation of Google's 2018 BERT model in PyTorch, allowing pre-training and fine-tuning for natural language processing tasks
Sentiment Analysis Tutorials
A tutorial repository for building sentiment analysis models with PyTorch using various architectures and techniques.
Segmentation Model Library
A comprehensive library for training and applying deep learning models for image segmentation
Hallucination mitigation
This project provides an official PyTorch implementation of a method to interpret and edit vision-language representations to mitigate hallucinations in image captions.
Featured in 4 awesome lists
Each link jumps to the spot where the list mentions captum.
