Alternatives and similar repositories for py-codegen:

Users that are interested in py-codegen are comparing it to the libraries listed below

• iree-org / iree-nvgpu
  ☆50Updated last year
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆84Updated this week
• pytorch-labs / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆102Updated this week
• cchan / tccl
  extensible collectives library in triton
  ☆85Updated last month
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆62Updated 2 weeks ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆106Updated 9 months ago
• makslevental / nelli
  A lightweight, Pythonic, frontend for MLIR
  ☆81Updated last year
• yifuwang / symm-mem-recipes
  ☆70Updated 4 months ago
• YashasSamaga / ConvolutionBuildingBlocks
  GEMM and Winograd based convolutions using CUTLASS
  ☆26Updated 4 years ago
• gunrock / loops
  🎃 GPU load-balancing library for regular and irregular computations.
  ☆62Updated 10 months ago
• sunlex0717 / DissectingTensorCores
  ☆96Updated last year
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆40Updated last month
• openxla / shardy
  MLIR-based partitioning system
  ☆82Updated this week
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆96Updated 9 months ago
• lixiuhong / batched_gemm
  ☆38Updated 5 years ago
• ColfaxResearch / cutlass-kernels
  ☆202Updated 9 months ago
• apuaaChen / EVT_AE
  Artifacts of EVT ASPLOS'24
  ☆24Updated last year
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆64Updated last year
• triton-lang / kernels
  ☆78Updated 6 months ago
• YulhwaKim / cutlass_tilesparse
  CUDA templates for tile-sparse matrix multiplication based on CUTLASS.
  ☆51Updated 7 years ago
• manya-bansal / mosaic
  ☆15Updated last year
• awslabs / ratex
  ☆23Updated 5 months ago
• enp1s0 / ozIMMU
  FP64 equivalent GEMM via Int8 Tensor Cores using the Ozaki scheme
  ☆60Updated last month
• Jokeren / triton-samples
  ☆27Updated 3 months ago
• facebookresearch / fairring
  Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …
  ☆65Updated 3 years ago
• ROCm / TransformerEngine
  ☆30Updated this week
• mayank31398 / cute-kernels
  A bunch of kernels that might make stuff slower 😉
  ☆40Updated this week
• uwsampl / SparseTIR
  SparseTIR: Sparse Tensor Compiler for Deep Learning
  ☆135Updated 2 years ago
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆38Updated 9 months ago
• cmu-catalyst / collage
  System for automated integration of deep learning backends.
  ☆48Updated 2 years ago