pytorchviz
by szagoruyko
A small package to create visualizations of PyTorch execution graphs
AI summary
Graph visualizer
A tool to visualize PyTorch execution graphs and traces
- stars
- 3.3K
- forks
- 281
- watching
- 31
Similar projects
Found by comparing what the projects do, not just their names.
GPU tensor library
A Python library providing tensors and dynamic neural networks with strong GPU acceleration
pytorch/examples22.5K
PyTorch examples
A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning.
Graph neural network library
A PyTorch-based library for training and applying Graph Neural Networks to structured data
diego999/pygat2.9K
Graph Network Model
An implementation of the Graph Attention Network model using PyTorch.
Deep Learning Template
An open-source PyTorch deep learning project template providing a structured environment for building and training various models.
Temporal graph NN library
A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models
Neural network trainer
A high-level library to help with training and evaluating neural networks in PyTorch
Deep Learning Tutorials
A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage.
nvidia/apex8.5K
Training toolkit
Tools for streamlined mixed precision and distributed training in PyTorch
Scene renderer
An implementation of NeRF in PyTorch for synthesizing novel views of complex scenes
Image Colorizer
A system for automatically colorizing black and white images with user interactions.
Feature visualization toolkit
Toolkit for visualizing neural network behavior in PyTorch
Scene Graph Parser
A software framework for scene graph parsing with global context using PyTorch and Visual Genome data.
Computer Vision Toolkit
A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision.
Translation model
Image-to-image translation in PyTorch using cycle-consistent adversarial networks and generative adversarial networks.
