google-research / sputnik

Related projects: ⓘ

• daadaada / turingas
  Assembler for NVIDIA Volta and Turing GPUs
  ☆195Updated 2 years ago
• ColfaxResearch / cutlass-kernels
  ☆138Updated 2 months ago
• KnowingNothing / MatmulTutorial
  A Easy-to-understand TensorOp Matmul Tutorial
  ☆265Updated this week
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆127Updated 2 years ago
• spcl / substation
  Research and development for optimizing transformers
  ☆121Updated 3 years ago
• owensgroup / merge-spmm
  Code for paper "Design Principles for Sparse Matrix Multiplication on the GPU" accepted to Euro-Par 2018
  ☆70Updated 3 years ago
• awslabs / raf
  ☆141Updated last year
• wangzyon / NVIDIA_SGEMM_PRACTICE
  Step-by-step optimization of CUDA SGEMM
  ☆207Updated 2 years ago
• sunlex0717 / DissectingTensorCores
  ☆73Updated 5 months ago
• Bruce-Lee-LY / cuda_hgemm
  Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…
  ☆264Updated last week
• microsoft / microxcaling
  PyTorch emulation library for Microscaling (MX)-compatible data formats
  ☆143Updated last month
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆52Updated 6 years ago
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆62Updated this week
• Cjkkkk / CUDA_gemm
  A simple high performance CUDA GEMM implementation.
  ☆319Updated 8 months ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆81Updated 2 months ago
• mmperf / mmperf
  MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.
  ☆123Updated 11 months ago
• mit-han-lab / inter-operator-scheduler
  [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration
  ☆191Updated 2 years ago
• albanD / subclass_zoo
  ☆136Updated 3 months ago
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆157Updated last week
• wzsh / wmma_tensorcore_sample
  Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)
  ☆109Updated 4 years ago
• Shigangli / Magicube
  Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.
  ☆79Updated last year
• leimao / CUDA-GEMM-Optimization
  CUDA Matrix Multiplication Optimization
  ☆118Updated 2 months ago
• YulhwaKim / cutlass_tilesparse
  CUDA templates for tile-sparse matrix multiplication based on CUTLASS.
  ☆48Updated 6 years ago
• uwsampl / SparseTIR
  SparseTIR: Sparse Tensor Compiler for Deep Learning
  ☆129Updated last year
• openxla / community
  Stores documents and resources used by the OpenXLA developer community
  ☆105Updated last month
• microsoft / SparTA
  ☆127Updated last month
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆250Updated this week
• openxla / openxla-nvgpu
  ☆48Updated 6 months ago
• microsoft / mscclpp
  MSCCL++: A GPU-driven communication stack for scalable AI applications
  ☆233Updated this week
• bytedance / flux
  A fast communication-overlapping library for tensor parallelism on GPUs.
  ☆184Updated this week