Skip to content

Deep-Learning-Profiling-Tools

Deep-Learning-Profiling-Tools is a collection of program analysis tools designed for GPU kernel development in deep learning workloads. The suite currently includes one tools:

  • Proton: A multi-level adaptive profiler for Triton that provides low-overhead profiling with custom metrics and SQL-like querying across different levels of abstraction.
  • Triton-Viz: A visualization and analysis toolkit for Triton that helps debug and understand kernel behavior by visualizing tensor operations and memory usage. It can run examples without requiring access to a GPU.

Together, these tools address the unique challenges of developing and optimizing GPU kernels in modern deep learning frameworks.

Popular repositories Loading

  1. triton-viz triton-vizPublic

    Python 353 32

  2. fasten fastenPublic

    Python 14 1

  3. triton-samples triton-samplesPublic

    Jupyter Notebook 14 3

  4. overhead-bench overhead-benchPublic

    Benchmarking profiler overhead

    Python 5 1

  5. linear-layout-viz linear-layout-vizPublic

    view triton linear layouts

    Python 5

  6. tensor-viz tensor-vizPublic

    N-dimensional tensor visualizer

    TypeScript 2 1

Repositories

Showing 10 of 19 repositories

Top languages

Loading…

Most used topics

Loading…