Alternatives and similar repositories for inference

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

• mlcommons / training
  Reference implementations of MLPerf® training benchmarks
  ☆1,715Updated last week
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,444Updated last week
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆921Updated this week
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,645Updated last week
• microsoft / nnfusion
  A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
  ☆994Updated last year
• intel / ai-reference-models
  Intel® AI Reference Models: contains Intel optimizations for running deep learning workloads on Intel® Xeon® Scalable processors and Inte…
  ☆719Updated this week
• NVIDIA / nccl-tests
  NCCL Tests
  ☆1,279Updated 3 weeks ago
• alibaba / BladeDISC
  BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.
  ☆895Updated 9 months ago
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆873Updated last week
• triton-inference-server / model_analyzer
  Triton Model Analyzer is a CLI tool to help with better understanding of the compute and memory requirements of the Triton Inference Serv…
  ☆491Updated 3 weeks ago
• ROCm / composable_kernel
  Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
  ☆471Updated this week
• pytorch / benchmark
  TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
  ☆989Updated this week
• NVIDIA / TensorRT-Model-Optimizer
  A unified library of state-of-the-art model optimization techniques like quantization, pruning, distillation, speculative decoding, etc. …
  ☆1,422Updated this week
• 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,500Updated this week
• NVIDIA / DCGM
  NVIDIA Data Center GPU Manager (DCGM) is a project for gathering telemetry and measuring the health of NVIDIA GPUs
  ☆586Updated last month
• 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,755Updated this week
• NVIDIA / nvbandwidth
  A tool for bandwidth measurements on NVIDIA GPUs.
  ☆536Updated 5 months ago
• onnx / optimizer
  ONNX Optimizer
  ☆757Updated 2 months ago
• tensorflow / mlir-hlo
  ☆421Updated this week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,063Updated last year
• pytorch / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,356Updated 2 weeks ago
• NVIDIA / nccl
  Optimized primitives for collective multi-GPU communication
  ☆4,108Updated last week
• msr-fiddle / pipedream
  ☆392Updated 2 years ago
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆728Updated last week
• ROCm / rccl
  ROCm Communication Collectives Library (RCCL)
  ☆386Updated this week
• d2l-ai / d2l-tvm
  Dive into Deep Learning Compiler
  ☆648Updated 3 years ago
• microsoft / superbenchmark
  A validation and profiling tool for AI infrastructure
  ☆337Updated this week
• intel / intel-extension-for-pytorch
  A Python package for extending the official PyTorch that can easily obtain performance on Intel platform
  ☆1,969Updated this week
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆730Updated 2 years ago
• flexflow / flexflow-train
  Automatically Discovering Fast Parallelization Strategies for Distributed Deep Neural Network Training
  ☆1,835Updated last month