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.
  ☆97Updated 2 months ago
• iree-org / iree-nvgpu
  ☆50Updated last year
• ademeure / cuda-side-boost
  ☆42Updated 4 months ago
• cchan / tccl
  extensible collectives library in triton
  ☆87Updated 5 months ago
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆76Updated last week
• openxla / shardy
  MLIR-based partitioning system
  ☆125Updated this week
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆271Updated 4 years ago
• meta-pytorch / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆45Updated 3 weeks ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆114Updated last year
• triton-lang / kernels
  ☆88Updated 10 months ago
• spcl / sten
  Sparsity support for PyTorch
  ☆37Updated 5 months ago
• makslevental / nelli
  A lightweight, Pythonic, frontend for MLIR
  ☆80Updated last year
• YashasSamaga / ConvolutionBuildingBlocks
  GEMM and Winograd based convolutions using CUTLASS
  ☆27Updated 5 years ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆354Updated this week
• ROCm / iris
  AMD RAD's experimental RMA library for Triton.
  ☆63Updated this week
• meta-pytorch / tritonparse
  TritonParse: A Compiler Tracer, Visualizer, and mini-Reproducer Generator(WIP) for Triton Kernels
  ☆150Updated this week
• hpcgarage / cuASR
  cuASR: CUDA Algebra for Semirings
  ☆38Updated 3 years ago
• Jokeren / triton-samples
  ☆28Updated 8 months ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆104Updated last year
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆134Updated 3 years ago
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆299Updated this week
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆41Updated last year
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆101Updated 7 months ago
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆53Updated last month
• awslabs / ratex
  ☆23Updated 3 weeks ago
• meta-pytorch / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆221Updated this week
• spcl / daceml
  A Data-Centric Compiler for Machine Learning
  ☆84Updated last year
• ColfaxResearch / cutlass-kernels
  ☆231Updated last year
• 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
• uwsampl / SparseTIR
  SparseTIR: Sparse Tensor Compiler for Deep Learning
  ☆138Updated 2 years ago