mlc-ai / notebooks

Related projects ⓘ

Alternatives and complementary repositories for notebooks

• KnowingNothing / MatmulTutorial
  A Easy-to-understand TensorOp Matmul Tutorial
  ☆290Updated 2 months ago
• ColfaxResearch / cutlass-kernels
  ☆167Updated 4 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.
  ☆29Updated 2 months ago
• 66RING / tiny-flash-attention
  flash attention tutorial written in python, triton, cuda, cutlass
  ☆202Updated 5 months ago
• tlc-pack / relax
  ☆196Updated last year
• AlibabaPAI / FLASHNN
  ☆79Updated 2 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…
  ☆302Updated 2 months ago
• apache / tvm-rfcs
  A home for the final text of all TVM RFCs.
  ☆101Updated last month
• OpenPPL / ppl.llm.kernel.cuda
  ☆138Updated 2 weeks ago
• Cjkkkk / CUDA_gemm
  A simple high performance CUDA GEMM implementation.
  ☆335Updated 10 months ago
• microsoft / SparTA
  ☆131Updated 3 months ago
• reed-lau / cute-gemm
  ☆79Updated 8 months ago
• LeiWang1999 / tvm_gpu_gemm
  play gemm with tvm
  ☆84Updated last year
• ColfaxResearch / cfx-article-src
  ☆48Updated this week
• MARD1NO / CUDA-PPT
  ☆79Updated last year
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆90Updated 4 months ago
• TiledTensor / TiledCUDA
  TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.
  ☆154Updated this week
• ConnollyLeon / awesome-Auto-Parallelism
  A baseline repository of Auto-Parallelism in Training Neural Networks
  ☆142Updated 2 years ago
• FlagOpen / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆219Updated 5 months ago
• DD-DuDa / Cute-Learning
  Examples of CUDA implementations by Cutlass CuTe
  ☆98Updated last week
• bytedance / flux
  A fast communication-overlapping library for tensor parallelism on GPUs.
  ☆224Updated 3 weeks ago
• OpenPPL / ppl.nn.llm
  ☆140Updated 6 months ago
• Yinghan-Li / YHs_Sample
  Yinghan's Code Sample
  ☆289Updated 2 years ago
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆191Updated this week
• mit-han-lab / parallel-computing-tutorial
  ☆143Updated last year
• thu-pacman / PET
  PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
  ☆114Updated 2 years ago
• tlc-pack / cutlass_fpA_intB_gemm
  A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer
  ☆85Updated 8 months ago
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆59Updated 6 years ago
• openmlsys / openmlsys-cuda
  Tutorials for writing high-performance GPU operators in AI frameworks.
  ☆123Updated last year
• microsoft / vattention
  Dynamic Memory Management for Serving LLMs without PagedAttention
  ☆238Updated last week