Alternatives and similar repositories for pytorch_open_registration_example

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

• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆109Updated last year
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆102Updated 7 years ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆112Updated last year
• apache / tvm-rfcs
  A home for the final text of all TVM RFCs.
  ☆109Updated last year
• mmperf / mmperf
  MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.
  ☆135Updated 2 years ago
• tlc-pack / TLCBench
  Benchmark scripts for TVM
  ☆74Updated 3 years ago
• cmu-catalyst / collage
  System for automated integration of deep learning backends.
  ☆47Updated 3 years ago
• awslabs / raf
  ☆145Updated 10 months ago
• OpenPPL / CuAssembler
  An unofficial cuda assembler, for all generations of SASS, hopefully ：）
  ☆84Updated 2 years ago
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆84Updated 2 weeks ago
• sunlex0717 / DissectingTensorCores
  ☆109Updated last year
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆221Updated this week
• tlc-pack / cutlass_fpA_intB_gemm
  A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer
  ☆96Updated 2 months ago
• ColfaxResearch / cfx-article-src
  ☆154Updated 6 months ago
• LeiWang1999 / tvm_gpu_gemm
  play gemm with tvm
  ☆92Updated 2 years ago
• AlibabaPAI / FLASHNN
  ☆102Updated last year
• daadaada / turingas
  Assembler for NVIDIA Volta and Turing GPUs
  ☆233Updated 3 years ago
• thu-pacman / PET
  PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
  ☆121Updated 3 years ago
• nox-410 / tvm.tl
  An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.
  ☆51Updated last year
• ROCm / TransformerEngine
  ☆51Updated this week
• apuaaChen / EVT_AE
  Artifacts of EVT ASPLOS'24
  ☆28Updated last year
• sjfeng1999 / gpu-arch-microbenchmark
  Dissecting NVIDIA GPU Architecture
  ☆110Updated 3 years ago
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆313Updated last month
• OpenPPL / ppl.llm.kernel.cuda
  ☆152Updated 10 months ago
• flagos-ai / flagtree
  FlagTree is a unified compiler for multiple AI chips, which is forked from triton-lang/triton.
  ☆137Updated this week
• intel / xetla
  ☆62Updated 11 months ago
• MARD1NO / CUDA-PPT
  ☆113Updated 7 months ago
• tlc-pack / relax
  ☆193Updated 2 years ago
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆115Updated this week
• iree-org / iree-nvgpu
  ☆50Updated last year