Alternatives and similar repositories for py-codegen

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

• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆90Updated 2 weeks ago
• iree-org / iree-nvgpu
  ☆50Updated last year
• ademeure / cuda-side-boost
  ☆20Updated 2 months ago
• openxla / shardy
  MLIR-based partitioning system
  ☆103Updated this week
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆70Updated this week
• pytorch-labs / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆187Updated this week
• cchan / tccl
  extensible collectives library in triton
  ☆87Updated 3 months ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆112Updated last year
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆51Updated last week
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆43Updated 3 months ago
• YashasSamaga / ConvolutionBuildingBlocks
  GEMM and Winograd based convolutions using CUTLASS
  ☆26Updated 5 years ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆99Updated last year
• Jokeren / triton-samples
  ☆28Updated 5 months ago
• makslevental / nelli
  A lightweight, Pythonic, frontend for MLIR
  ☆80Updated last year
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆270Updated 4 years ago
• triton-lang / kernels
  ☆83Updated 8 months ago
• sunlex0717 / DissectingTensorCores
  ☆102Updated last year
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆79Updated last week
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆184Updated this week
• uwsampl / SparseTIR
  SparseTIR: Sparse Tensor Compiler for Deep Learning
  ☆138Updated 2 years ago
• sandeepkumar-skb / pytorch_custom_op
  End to End steps for adding custom ops in PyTorch.
  ☆23Updated 4 years ago
• spcl / daceml
  A Data-Centric Compiler for Machine Learning
  ☆84Updated last year
• hpcgarage / cuASR
  cuASR: CUDA Algebra for Semirings
  ☆36Updated 2 years ago
• ColfaxResearch / cutlass-kernels
  ☆216Updated last year
• open-lm-engine / cute-kernels
  A bunch of kernels that might make stuff slower 😉
  ☆54Updated this week
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆40Updated 11 months ago
• awslabs / ratex
  ☆23Updated 7 months ago
• oresths / tSparse
  A GPU algorithm for sparse matrix-matrix multiplication
  ☆71Updated 4 years ago
• nod-ai / SHARK-ModelDev
  Unified compiler/runtime for interfacing with PyTorch Dynamo.
  ☆100Updated last week
• spcl / sten
  Sparsity support for PyTorch
  ☆35Updated 3 months ago