Alternatives and similar repositories for notebooks:

Users that are interested in notebooks are comparing it to the libraries listed below

• KnowingNothing / MatmulTutorial
  A Easy-to-understand TensorOp Matmul Tutorial
  ☆337Updated 6 months ago
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆236Updated last week
• Cjkkkk / CUDA_gemm
  A simple high performance CUDA GEMM implementation.
  ☆357Updated last year
• TiledTensor / TiledCUDA
  We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …
  ☆180Updated 2 months ago
• Bruce-Lee-LY / flash_attention_inference
  Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.
  ☆35Updated last month
• ColfaxResearch / cutlass-kernels
  ☆193Updated 8 months ago
• AyakaGEMM / Hands-on-GEMM
  ☆115Updated last year
• Cambricon / triton-linalg
  Development repository for the Triton-Linalg conversion
  ☆182Updated last month
• tlc-pack / relax
  ☆193Updated 2 years ago
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆87Updated 6 years ago
• ConnollyLeon / awesome-Auto-Parallelism
  A baseline repository of Auto-Parallelism in Training Neural Networks
  ☆143Updated 2 years ago
• reed-lau / cute-gemm
  ☆112Updated 3 months ago
• Bruce-Lee-LY / cuda_hgemm
  Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…
  ☆374Updated 6 months ago
• OpenPPL / ppl.llm.kernel.cuda
  ☆145Updated 2 months ago
• Yinghan-Li / YHs_Sample
  Yinghan's Code Sample
  ☆316Updated 2 years ago
• ColfaxResearch / cfx-article-src
  ☆94Updated 3 weeks ago
• Guangxuan-Xiao / torch-int
  This repository contains integer operators on GPUs for PyTorch.
  ☆198Updated last year
• FlagOpen / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆257Updated 10 months ago
• LeiWang1999 / tvm_gpu_gemm
  play gemm with tvm
  ☆89Updated last year
• leimao / CUDA-GEMM-Optimization
  CUDA Matrix Multiplication Optimization
  ☆178Updated 8 months ago
• mit-han-lab / parallel-computing-tutorial
  ☆160Updated last year
• DD-DuDa / Cute-Learning
  Examples of CUDA implementations by Cutlass CuTe
  ☆150Updated 2 months ago
• openmlsys / openmlsys-cuda
  Tutorials for writing high-performance GPU operators in AI frameworks.
  ☆130Updated last year
• AlibabaPAI / FLASHNN
  ☆91Updated 6 months ago
• yzhaiustc / Optimizing-SGEMM-on-NVIDIA-Turing-GPUs
  Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.
  ☆334Updated 3 months ago
• 66RING / tiny-flash-attention
  flash attention tutorial written in python, triton, cuda, cutlass
  ☆315Updated 3 months ago
• thu-pacman / PET
  PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
  ☆119Updated 2 years ago
• microsoft / SparTA
  ☆136Updated 8 months ago
• apache / tvm-rfcs
  A home for the final text of all TVM RFCs.
  ☆102Updated 6 months ago
• uwsampl / SparseTIR
  SparseTIR: Sparse Tensor Compiler for Deep Learning
  ☆134Updated 2 years ago