1 repository
Tools that provide visual representations of neural network graphs to analyze compiler optimizations.
Distinguishing note: Specifically visualizes compiler-driven graph optimizations rather than general model compression or financial modeling.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · Model Optimization Visualizers. Refine with filters or upvote what's useful.
AITemplate is an ahead-of-time deep learning compiler that translates PyTorch neural networks into standalone C++ source code. It functions as a PyTorch to C++ compiler and a GPU kernel fusion engine, producing self-contained executable binaries that run inference without requiring a Python interpreter or deep learning framework runtime. The project generates optimized CUDA and HIP C++ code specifically for NVIDIA TensorCores and AMD MatrixCores. It focuses on maximizing throughput for half-precision floating-point operations through a system that combines multiple neural network operators in
Generates web-based representations of the model structure to illustrate how the generation process optimizes the network.