Alternatives and similar repositories for My-Torch-Extension

Users that are interested in My-Torch-Extension are comparing it to the libraries listed below

• ifromeast / cuda_learning
  learning how CUDA works
  ☆295Updated 5 months ago
• 66RING / tiny-flash-attention
  flash attention tutorial written in python, triton, cuda, cutlass
  ☆402Updated 2 months ago
• AyakaGEMM / Hands-on-GEMM
  ☆137Updated last year
• harleyszhang / llm_counts
  llm theoretical performance analysis tools and support params, flops, memory and latency analysis.
  ☆101Updated 3 weeks ago
• harleyszhang / lite_llama
  A light llama-like llm inference framework based on the triton kernel.
  ☆144Updated last week
• xlite-dev / ffpa-attn
  ⚡️FFPA: Extend FlashAttention-2 with Split-D, achieve ~O(1) SRAM complexity for large headdim, 1.8x~3x↑ vs SDPA.🎉
  ☆194Updated 2 months ago
• DD-DuDa / Cute-Learning
  Examples of CUDA implementations by Cutlass CuTe
  ☆214Updated last month
• FlagOpen / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆275Updated last year
• SiriusNEO / Triton-Puzzles-Lite
  Puzzles for learning Triton, play it with minimal environment configuration!
  ☆446Updated 8 months ago
• mdy666 / mdy_triton
  ☆140Updated last month
• nicolaswilde / cuda-sgemm
  ☆67Updated 7 months ago
• reed-lau / cute-gemm
  ☆128Updated 8 months ago
• RussWong / CUDATutorial
  A CUDA tutorial to make people learn CUDA program from 0
  ☆247Updated last year
• luliyucoordinate / cute-flash-attention
  Implement Flash Attention using Cute.
  ☆92Updated 7 months ago
• KnowingNothing / MatmulTutorial
  A Easy-to-understand TensorOp Matmul Tutorial
  ☆370Updated 10 months ago
• YuxueYang1204 / CudaDemo
  Implement custom operators in PyTorch with cuda/c++
  ☆66Updated 2 years ago
• weishengying / tiny-flash-attention
  使用 cutlass 实现 flash-attention 精简版，具有教学意义
  ☆45Updated 11 months ago
• hyperai / triton-cn
  Triton Documentation in Chinese Simplified / Triton 中文文档
  ☆78Updated 3 months ago
• BBuf / how-to-learn-deep-learning-framework
  how to learn PyTorch and OneFlow
  ☆445Updated last year
• Cjkkkk / CUDA_gemm
  A simple high performance CUDA GEMM implementation.
  ☆392Updated last year
• zjhellofss / triton_course
  ☆33Updated 2 months ago
• CalebDu / Awesome-Cute
  ☆91Updated 2 months ago
• ruikangliu / FlatQuant
  [ICML 2025] Official PyTorch implementation of "FlatQuant: Flatness Matters for LLM Quantization"
  ☆151Updated 2 weeks ago
• madsys-dev / deepseekv2-profile
  ☆145Updated 5 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.
  ☆39Updated 5 months ago
• RussWong / LLM-engineering
  ☆24Updated 4 months ago
• weishengying / cutlass_flash_atten_fp8
  使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention
  ☆74Updated 11 months ago
• mit-han-lab / parallel-computing-tutorial
  ☆171Updated last year
• li199603 / parallel_prefix_sum
  Parallel Prefix Sum (Scan) with CUDA
  ☆24Updated last year
• OpenPPL / ppl.pmx
  ☆59Updated 8 months ago