Alternatives and similar repositories for FBGEMM

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

• dmlc / dlpack
  common in-memory tensor structure
  ☆1,120Updated last week
• microsoft / nnfusion
  A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
  ☆1,001Updated last year
• tensorflow / runtime
  A performant and modular runtime for TensorFlow
  ☆756Updated 3 months ago
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆731Updated 2 years ago
• pytorch / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,378Updated 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,700Updated this week
• alibaba / BladeDISC
  BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.
  ☆910Updated 11 months ago
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆953Updated last week
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆907Updated this week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,069Updated last year
• NVIDIA / TransformerEngine
  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on H…
  ☆3,007Updated this week
• tensorflow / mlir-hlo
  ☆422Updated this week
• d2l-ai / d2l-tvm
  Dive into Deep Learning Compiler
  ☆645Updated 3 years ago
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆773Updated last week
• google / gemmlowp
  Low-precision matrix multiplication
  ☆1,821Updated last year
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆367Updated this week
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆579Updated last week
• ROCm / composable_kernel
  Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
  ☆499Updated this week
• openai / blocksparse
  Efficient GPU kernels for block-sparse matrix multiplication and convolution
  ☆1,063Updated 2 years ago
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆280Updated this week
• tensorflow / custom-op
  Guide for building custom op for TensorFlow
  ☆385Updated 2 years ago
• mlcommons / training
  Reference implementations of MLPerf® training benchmarks
  ☆1,734Updated last week
• pytorch / benchmark
  TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
  ☆1,003Updated this week
• mlcommons / inference
  Reference implementations of MLPerf® inference benchmarks
  ☆1,506Updated this week
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆737Updated 3 months ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆509Updated 4 years ago
• ByteDance-Seed / Triton-distributed
  Distributed Compiler based on Triton for Parallel Systems
  ☆1,280Updated this week
• NVIDIA / cub
  [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
  ☆1,809Updated 2 years ago
• bytedance / effective_transformer
  Running BERT without Padding
  ☆476Updated 3 years ago
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆449Updated last week