Alternatives and similar repositories for FBGEMM

Users that are interested in FBGEMM are comparing it to the libraries listed below

• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆715Updated 2 years ago
• microsoft / nnfusion
  A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
  ☆989Updated 9 months ago
• dmlc / dlpack
  common in-memory tensor structure
  ☆1,014Updated last week
• tensorflow / runtime
  A performant and modular runtime for TensorFlow
  ☆762Updated 2 months ago
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,562Updated last week
• pytorch / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,315Updated this week
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆873Updated last week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,051Updated last year
• alibaba / BladeDISC
  BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.
  ☆873Updated 5 months ago
• d2l-ai / d2l-tvm
  Dive into Deep Learning Compiler
  ☆644Updated 3 years ago
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆666Updated last week
• NVIDIA / cub
  [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
  ☆1,755Updated last year
• tensorflow / mlir-hlo
  ☆416Updated this week
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆703Updated this week
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆821Updated this week
• NVIDIA / TransformerEngine
  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Bla…
  ☆2,491Updated this week
• pytorch / tvm
  TVM integration into PyTorch
  ☆453Updated 5 years ago
• google / gemmlowp
  Low-precision matrix multiplication
  ☆1,805Updated last year
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆490Updated this week
• onnx / optimizer
  ONNX Optimizer
  ☆721Updated last week
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆258Updated 2 years ago
• mlcommons / inference
  Reference implementations of MLPerf™ inference benchmarks
  ☆1,397Updated last week
• openai / blocksparse
  Efficient GPU kernels for block-sparse matrix multiplication and convolution
  ☆1,041Updated 2 years ago
• tpoisonooo / how-to-optimize-gemm
  row-major matmul optimization
  ☆637Updated last year
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆503Updated 3 years ago
• msr-fiddle / pipedream
  ☆391Updated 2 years ago
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆387Updated last week
• libxsmm / libxsmm
  Library for specialized dense and sparse matrix operations, and deep learning primitives.
  ☆879Updated 2 weeks ago
• NVIDIA / cuCollections
  ☆542Updated last week
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆588Updated this week