Alternatives and similar repositories for dlpack

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

• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,487Updated last week
• pytorch / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,372Updated this week
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆731Updated 2 years ago
• tensorflow / runtime
  A performant and modular runtime for TensorFlow
  ☆757Updated 3 months ago
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,068Updated last year
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆275Updated 3 weeks ago
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆897Updated last week
• microsoft / nnfusion
  A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
  ☆995Updated last year
• tensorflow / mlir-hlo
  ☆422Updated last week
• NVIDIA / cuCollections
  ☆597Updated this week
• zdevito / ATen
  ATen: A TENsor library for C++11
  ☆710Updated 6 years ago
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆663Updated this week
• pytorch / tvm
  TVM integration into PyTorch
  ☆455Updated 5 years ago
• tensor-compiler / taco
  The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
  ☆1,332Updated 7 months ago
• tensorflow / custom-op
  Guide for building custom op for TensorFlow
  ☆385Updated 2 years ago
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆576Updated this week
• d2l-ai / d2l-tvm
  Dive into Deep Learning Compiler
  ☆646Updated 3 years ago
• inducer / loopy
  A code generator for array-based code on CPUs and GPUs
  ☆617Updated last week
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,684Updated this week
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆736Updated 3 months ago
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆940Updated last week
• pytorch / builder
  Continuous builder and binary build scripts for pytorch
  ☆356Updated 3 months ago
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆772Updated 2 weeks ago
• BaguaSys / bagua
  Bagua Speeds up PyTorch
  ☆881Updated last year
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆437Updated 2 weeks ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆509Updated 4 years ago
• pytorch / benchmark
  TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
  ☆998Updated this week
• alibaba / BladeDISC
  BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.
  ☆907Updated 11 months ago
• libxsmm / libxsmm
  Library for specialized dense and sparse matrix operations, and deep learning primitives.
  ☆922Updated last week
• NVIDIA / cub
  [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
  ☆1,807Updated 2 years ago