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
  ☆207Updated 6 months ago
• harleyszhang / llm_counts
  llm theoretical performance analysis tools and support params, flops, memory and latency analysis.
  ☆79Updated last month
• RussWong / CUDATutorial
  A CUDA tutorial to make people learn CUDA program from 0
  ☆212Updated 7 months ago
• 66RING / tiny-flash-attention
  flash attention tutorial written in python, triton, cuda, cutlass
  ☆286Updated 2 months ago
• luliyucoordinate / cute-flash-attention
  Implement Flash Attention using Cute.
  ☆70Updated 2 months ago
• DD-DuDa / Cute-Learning
  Examples of CUDA implementations by Cutlass CuTe
  ☆141Updated last month
• harleyszhang / lite_llama
  A light llama-like llm inference framework based on the triton kernel.
  ☆92Updated last week
• DefTruth / ffpa-attn-mma
  📚FFPA(Split-D): Yet another Faster Flash Prefill Attention with O(1)⚡️GPU SRAM complexity for headdim > 256, ~2x↑🎉vs SDPA EA.
  ☆123Updated last week
• AyakaGEMM / Hands-on-GEMM
  ☆111Updated 11 months ago
• reed-lau / cute-gemm
  ☆101Updated 2 months ago
• openmlsys / openmlsys-cuda
  Tutorials for writing high-performance GPU operators in AI frameworks.
  ☆129Updated last year
• SiriusNEO / Triton-Puzzles-Lite
  Puzzles for learning Triton, play it with minimal environment configuration!
  ☆248Updated 3 months ago
• hyperai / triton-cn
  Triton Documentation in Chinese Simplified / Triton 中文文档
  ☆56Updated last month
• caiwanxianhust / FasterLLaMA
  使用 CUDA C++ 实现的 llama 模型推理框架
  ☆46Updated 3 months ago
• weishengying / tiny-flash-attention
  使用 cutlass 实现 flash-attention 精简版，具有教学意义
  ☆36Updated 6 months ago
• madsys-dev / deepseekv2-profile
  ☆116Updated this week
• DefTruth / hgemm-tensorcores-mma
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆53Updated last week
• chenhongyu2048 / LLM-inference-optimization-paper
  Summary of some awesome work for optimizing LLM inference
  ☆58Updated 3 weeks ago
• RussWong / LLM-engineering
  ☆18Updated 9 months ago
• YuxueYang1204 / CudaDemo
  Implement custom operators in PyTorch with cuda/c++
  ☆54Updated 2 years ago
• CalvinXKY / BasicCUDA
  A tutorial for CUDA&PyTorch
  ☆127Updated last month
• KnowingNothing / MatmulTutorial
  A Easy-to-understand TensorOp Matmul Tutorial
  ☆322Updated 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.
  ☆35Updated this week
• CalebDu / Awesome-Cute
  ☆39Updated last month
• Tongkaio / CUDA_Kernel_Samples
  CUDA 算子手撕与面试指南
  ☆174Updated last month
• weishengying / cutlass_flash_atten_fp8
  使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention
  ☆54Updated 6 months ago
• OpenPPL / ppl.pmx
  ☆58Updated 3 months ago
• iclementine / optimize_softmax
  Optimize softmax in triton in many cases
  ☆18Updated 5 months ago
• Bruce-Lee-LY / decoding_attention
  Decoding Attention is specially optimized for multi head attention (MHA) using CUDA core for the decoding stage of LLM inference.
  ☆29Updated this week
• mdy666 / mdy_triton
  ☆71Updated this week