mlc-ai / notebooks

Related projects: ⓘ

• KnowingNothing / MatmulTutorial
  A Easy-to-understand TensorOp Matmul Tutorial
  ☆265Updated this week
• tlc-pack / relax
  ☆193Updated last year
• FlagOpen / FlagGems
  FlagGems is an operator library for large language models implemented in Triton Language.
  ☆246Updated this week
• OpenPPL / ppl.llm.kernel.cuda
  ☆133Updated 2 months ago
• 66RING / tiny-flash-attention
  flash attention tutorial written in python, triton, cuda, cutlass
  ☆159Updated 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…
  ☆264Updated last week
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆157Updated last week
• Cjkkkk / CUDA_gemm
  A simple high performance CUDA GEMM implementation.
  ☆319Updated 8 months ago
• bytedance / flux
  A fast communication-overlapping library for tensor parallelism on GPUs.
  ☆184Updated this week
• apache / tvm-rfcs
  A home for the final text of all TVM RFCs.
  ☆99Updated 3 months ago
• ColfaxResearch / cutlass-kernels
  ☆138Updated 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.
  ☆20Updated last week
• FlagOpen / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆204Updated 3 months ago
• ConnollyLeon / awesome-Auto-Parallelism
  A baseline repository of Auto-Parallelism in Training Neural Networks
  ☆138Updated 2 years ago
• openppl-public / ppl.nn.llm
  ☆140Updated 4 months ago
• LeiWang1999 / tvm_gpu_gemm
  play gemm with tvm
  ☆81Updated last year
• Cambricon / triton-linalg
  Development repository for the Triton-Linalg conversion
  ☆135Updated last month
• thu-pacman / PET
  PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
  ☆112Updated 2 years ago
• Yinghan-Li / YHs_Sample
  Yinghan's Code Sample
  ☆272Updated 2 years ago
• microsoft / vattention
  Dynamic Memory Management for Serving LLMs without PagedAttention
  ☆185Updated last month
• mit-han-lab / parallel-computing-tutorial
  ☆134Updated last year
• reed-lau / cute-gemm
  ☆69Updated 6 months ago
• AlibabaPAI / FLASHNN
  ☆67Updated last week
• MARD1NO / CUDA-PPT
  ☆77Updated last year
• microsoft / SparTA
  ☆127Updated last month
• mit-han-lab / inter-operator-scheduler
  [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration
  ☆191Updated 2 years ago
• yzhaiustc / Optimizing-SGEMM-on-NVIDIA-Turing-GPUs
  Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.
  ☆265Updated 2 years ago
• bytedance / byteir
  A model compilation solution for various hardware
  ☆357Updated this week
• openmlsys / openmlsys-cuda
  Tutorials for writing high-performance GPU operators in AI frameworks.
  ☆118Updated last year
• ifromeast / cuda_learning
  learning how CUDA works
  ☆149Updated last month