csarofeen / pytorch

Related projects ⓘ

Alternatives and complementary repositories for pytorch

• zasdfgbnm / autonvtx
  Automatically insert nvtx ranges to PyTorch models
  ☆17Updated 3 years ago
• jansel / pytorch-jit-paritybench
  ☆36Updated 5 months ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆268Updated this week
• iree-org / iree-nvgpu
  ☆48Updated 8 months ago
• bwasti / pytorch_compiler_tutorial
  Codebase associated with the PyTorch compiler tutorial
  ☆44Updated 5 years ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆86Updated last week
• openxla / shardy
  MLIR-based partitioning system
  ☆36Updated this week
• microsoft / torchy
  A tracing JIT for PyTorch
  ☆17Updated 2 years ago
• pytorch / rfcs
  PyTorch RFCs (experimental)
  ☆127Updated 2 months ago
• makslevental / nelli
  A lightweight, Pythonic, frontend for MLIR
  ☆79Updated last year
• YashasSamaga / ConvolutionBuildingBlocks
  GEMM and Winograd based convolutions using CUTLASS
  ☆25Updated 4 years ago
• mmperf / mmperf
  MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.
  ☆123Updated last year
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆143Updated this week
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆185Updated this week
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆30Updated 3 months ago
• ROCm / rocMLIR
  ☆128Updated this week
• 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
• albanD / subclass_zoo
  ☆147Updated 4 months ago
• masahi / torchscript-to-tvm
  ☆67Updated last year
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆82Updated 3 months ago
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆28Updated 5 months ago
• mlcommons / inference_policies
  Issues related to MLPerf™ Inference policies, including rules and suggested changes
  ☆57Updated this week
• manishucsd / py-codegen
  ☆14Updated last month
• intel / xetla
  ☆57Updated this week
• awslabs / raf
  ☆140Updated last year
• NVIDIA / compute-sanitizer-samples
  Samples demonstrating how to use the Compute Sanitizer Tools and Public API
  ☆65Updated last year
• MegEngine / cutlass-bak
  modified cutlass
  ☆14Updated 4 years ago
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆59Updated 6 years ago
• ROCm / Tensile
  Stretching GPU performance for GEMMs and tensor contractions.
  ☆220Updated this week