Alternatives and similar repositories for FBGEMM

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

• tensorflow / runtime
  A performant and modular runtime for TensorFlow
  ☆759Updated last week
• dmlc / dlpack
  common in-memory tensor structure
  ☆1,058Updated last week
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆731Updated 2 years ago
• microsoft / nnfusion
  A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
  ☆994Updated 11 months ago
• pytorch / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,352Updated 3 weeks ago
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,627Updated this week
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆911Updated this week
• alibaba / BladeDISC
  BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.
  ☆891Updated 8 months ago
• tensorflow / mlir-hlo
  ☆419Updated last week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,062Updated last year
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆861Updated last week
• d2l-ai / d2l-tvm
  Dive into Deep Learning Compiler
  ☆647Updated 3 years ago
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆532Updated last week
• msr-fiddle / pipedream
  ☆393Updated 2 years ago
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆265Updated 3 weeks ago
• openai / blocksparse
  Efficient GPU kernels for block-sparse matrix multiplication and convolution
  ☆1,049Updated 2 years ago
• ROCm / composable_kernel
  Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
  ☆455Updated this week
• pytorch / tvm
  TVM integration into PyTorch
  ☆453Updated 5 years ago
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆721Updated last week
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆714Updated last week
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆271Updated 4 years ago
• google / gemmlowp
  Low-precision matrix multiplication
  ☆1,815Updated last year
• tensor-compiler / taco
  The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
  ☆1,323Updated 4 months ago
• NVIDIA / cudnn-frontend
  cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it
  ☆612Updated last 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,720Updated this week
• pytorch / PiPPy
  Pipeline Parallelism for PyTorch
  ☆779Updated last year
• kakaobrain / torchgpipe
  A GPipe implementation in PyTorch
  ☆854Updated last year
• pytorch / benchmark
  TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
  ☆987Updated this week
• baidu-research / baidu-allreduce
  ☆593Updated 7 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆505Updated 4 years ago