TylerYep/torchinfo

A modern model graph visualizer and debugger

ML models and internal tensors 3D visualizer

Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-CAM, Finer-CAM, LeGrad, RefineCAM)

Model interpretability and understanding for PyTorch

Visualizer for neural network, deep learning and machine learning models

Tools to Design or Visualize Architecture of Neural Network

A JAX research toolkit for building, editing, and visualizing neural networks.

Collection of common code that's shared among different research projects in FAIR computer vision team.

Model analysis tools for TensorFlow