NVIDIA / cutlass

Related projects: ⓘ

• NVIDIA / cuda-samples
  Samples for CUDA Developers which demonstrates features in CUDA Toolkit
  ☆6,087Updated last month
• openxla / xla
  A machine learning compiler for GPUs, CPUs, and ML accelerators
  ☆2,577Updated this week
• NVIDIA / CUDALibrarySamples
  CUDA Library Samples
  ☆1,519Updated last week
• NVIDIA / nccl
  Optimized primitives for collective multi-GPU communication
  ☆3,132Updated this week
• NVIDIA / cub
  [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
  ☆1,669Updated 11 months ago
• oneapi-src / oneDNN
  oneAPI Deep Neural Network Library (oneDNN)
  ☆3,579Updated this week
• NVIDIA / cccl
  CUDA Core Compute Libraries
  ☆1,132Updated this week
• NVIDIA / FasterTransformer
  Transformer related optimization, including BERT, GPT
  ☆5,773Updated 5 months ago
• NVIDIA / TransformerEngine
  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…
  ☆1,811Updated this week
• iree-org / iree
  A retargetable MLIR-based machine learning compiler and runtime toolkit.
  ☆2,559Updated this week
• NVIDIA / TensorRT
  NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…
  ☆10,552Updated last week
• triton-lang / triton
  Development repository for the Triton language and compiler
  ☆12,698Updated this week
• NVIDIA / thrust
  [ARCHIVED] The C++ parallel algorithms library. See https://github.com/NVIDIA/cccl
  ☆4,907Updated 7 months ago
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,301Updated this week
• flame / how-to-optimize-gemm
  ☆1,725Updated last year
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,169Updated this week
• apache / tvm
  Open deep learning compiler stack for cpu, gpu and specialized accelerators
  ☆11,602Updated this week
• google / XNNPACK
  High-efficiency floating-point neural network inference operators for mobile, server, and Web
  ☆1,812Updated this week
• cuda-mode / lectures
  Material for cuda-mode lectures
  ☆2,401Updated 2 weeks ago
• NVIDIA / MatX
  An efficient C++17 GPU numerical computing library with Python-like syntax
  ☆1,187Updated last week
• google / gemmlowp
  Low-precision matrix multiplication
  ☆1,772Updated 7 months ago
• intel / neural-compressor
  SOTA low-bit LLM quantization (INT8/FP8/INT4/FP4/NF4) & sparsity; leading model compression techniques on TensorFlow, PyTorch, and ONNX R…
  ☆2,146Updated this week
• pytorch / TensorRT
  PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
  ☆2,499Updated this week
• halide / Halide
  a language for fast, portable data-parallel computation
  ☆5,835Updated this week
• pytorch / glow
  Compiler for Neural Network hardware accelerators
  ☆3,206Updated 4 months ago
• facebookincubator / AITemplate
  AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…
  ☆4,531Updated this week
• arrayfire / arrayfire
  ArrayFire: a general purpose GPU library.
  ☆4,525Updated last week
• HazyResearch / ThunderKittens
  Tile primitives for speedy kernels
  ☆1,489Updated this week
• cupy / cupy
  NumPy & SciPy for GPU
  ☆8,124Updated this week
• facebookresearch / fairscale
  PyTorch extensions for high performance and large scale training.
  ☆3,149Updated 2 weeks ago