manishucsd / py-codegen

Related projects ⓘ

Alternatives and complementary repositories for py-codegen

• iree-org / iree-nvgpu
  ☆48Updated 8 months ago
• YashasSamaga / ConvolutionBuildingBlocks
  GEMM and Winograd based convolutions using CUTLASS
  ☆25Updated 4 years ago
• cchan / tccl
  extensible collectives library in triton
  ☆72Updated last month
• facebookresearch / fairring
  Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …
  ☆63Updated 2 years ago
• hpcgarage / cuASR
  cuASR: CUDA Algebra for Semirings
  ☆34Updated 2 years ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆60Updated 8 months ago
• brightlaboratory / polydl
  ☆11Updated 3 years ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆90Updated 4 months ago
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆35Updated 6 months ago
• ColfaxResearch / cutlass-kernels
  ☆169Updated 4 months ago
• sunlex0717 / DissectingTensorCores
  ☆80Updated 7 months ago
• triton-lang / kernels
  ☆45Updated 2 weeks ago
• ColfaxResearch / cfx-article-src
  ☆48Updated this week
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆84Updated 4 months ago
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆28Updated 5 months ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last month
• makslevental / nelli
  A lightweight, Pythonic, frontend for MLIR
  ☆80Updated last year
• enp1s0 / ozIMMU
  FP64 equivalent GEMM via Int8 Tensor Cores using the Ozaki scheme
  ☆46Updated 2 months ago
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆249Updated 3 years ago
• yifuwang / symm-mem-recipes
  ☆12Updated last month
• UDC-GAC / openCNN
  A Winograd Minimal Filter Implementation in CUDA
  ☆23Updated 3 years ago
• UDC-GAC / venom
  A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores
  ☆43Updated 11 months ago
• Bruce-Lee-LY / cuda_back2back_hgemm
  Use tensor core to calculate back-to-back HGEMM (half-precision general matrix multiplication) with MMA PTX instruction.
  ☆11Updated last year
• lixiuhong / batched_gemm
  ☆38Updated 4 years ago
• YulhwaKim / cutlass_tilesparse
  CUDA templates for tile-sparse matrix multiplication based on CUTLASS.
  ☆49Updated 6 years ago
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆30Updated 3 months ago
• gunrock / loops
  🎃 GPU load-balancing library for regular and irregular computations.
  ☆57Updated 5 months ago
• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago
• spcl / sten
  Sparsity support for PyTorch
  ☆31Updated this week
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago