pytorch-labs / triton-cpu

Related projects ⓘ

Alternatives and complementary repositories for triton-cpu

• triton-lang / triton-cpu
  An experimental CPU backend for Triton
  ☆56Updated last week
• cchan / tccl
  extensible collectives library in triton
  ☆71Updated last month
• triton-lang / kernels
  ☆45Updated 2 weeks ago
• ColfaxResearch / cfx-article-src
  ☆48Updated this week
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆191Updated this week
• ROCm / rocMLIR
  ☆128Updated this week
• nod-ai / SHARK-ModelDev
  Unified compiler/runtime for interfacing with PyTorch Dynamo.
  ☆95Updated last week
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆84Updated 4 months ago
• iree-org / iree-turbine
  IREE's PyTorch Frontend, based on Torch Dynamo.
  ☆55Updated this week
• mlc-ai / xgrammar
  Efficient, Flexible and Portable Structured Generation
  ☆40Updated this week
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆143Updated this week
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆90Updated 4 months ago
• intel / torch-xpu-ops
  ☆29Updated this week
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆93Updated this week
• iree-org / iree-nvgpu
  ☆48Updated 8 months ago
• ColfaxResearch / cutlass-kernels
  ☆167Updated 4 months ago
• TiledTensor / TiledCUDA
  TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.
  ☆154Updated this week
• openxla / shardy
  MLIR-based partitioning system
  ☆40Updated this week
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆68Updated 3 weeks ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆19Updated this week
• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago
• mobiusml / gemlite
  Simple and fast low-bit matmul kernels in CUDA / Triton
  ☆143Updated this week
• leimao / CUTLASS-Examples
  CUTLASS and CuTe Examples
  ☆13Updated this week
• plaidml / tpp-mlir
  TPP experimentation on MLIR for linear algebra
  ☆110Updated last month
• uwsampl / SparseTIR
  SparseTIR: Sparse Tensor Compiler for Deep Learning
  ☆131Updated last year
• iree-org / iree-experimental
  Experiments and prototypes associated with IREE or MLIR
  ☆49Updated 3 months ago
• intel / mlir-extensions
  Intel® Extension for MLIR. A staging ground for MLIR dialects and tools for Intel devices using the MLIR toolchain.
  ☆124Updated this week
• sunlex0717 / DissectingTensorCores
  ☆80Updated 7 months ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆270Updated this week
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆98Updated 2 months ago