pytorchviz
Graph visualizer
A tool to visualize PyTorch execution graphs and traces
A small package to create visualizations of PyTorch execution graphs
3k stars
31 watching
279 forks
Language: Jupyter Notebook
last commit: 8 months ago Related projects:
| Repository | Description | Stars |
|---|---|---|
 pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 84,371 |
| pytorch/examples | A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning. | 22,428 |
 pyg-team/pytorch_geometric | A PyTorch-based library for training and applying Graph Neural Networks to structured data | 21,382 |
 diego999/pygat | An implementation of the Graph Attention Network model using PyTorch. | 2,914 |
 victoresque/pytorch-template | An open-source PyTorch deep learning project template providing a structured environment for building and training various models. | 4,754 |
 benedekrozemberczki/pytorch_geometric_temporal | A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models | 2,677 |
| pytorch/ignite | A high-level library to help with training and evaluating neural networks in PyTorch | 4,531 |
 yunjey/pytorch-tutorial | A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage. | 30,286 |
 nvidia/apex | Tools for streamlined mixed precision and distributed training in PyTorch | 8,417 |
 yenchenlin/nerf-pytorch | An implementation of NeRF in PyTorch for synthesizing novel views of complex scenes | 5,505 |
 junyanz/interactive-deep-colorization | A system for automatically colorizing black and white images with user interactions. | 2,694 |
 misaogura/flashtorch | Toolkit for visualizing neural network behavior in PyTorch | 734 |
 rowanz/neural-motifs | A software framework for scene graph parsing with global context using PyTorch and Visual Genome data. | 525 |
 kaiyangzhou/dassl.pytorch | A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. | 1,217 |
| junyanz/pytorch-cyclegan-and-pix2pix | Image-to-image translation in PyTorch using cycle-consistent adversarial networks and generative adversarial networks. | 23,085 |