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,399Updated this week
• tensorflow / runtime
  A performant and modular runtime for TensorFlow
  ☆758Updated 2 months ago
• pytorch / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,326Updated 3 weeks ago
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆719Updated 2 years ago
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆259Updated 2 years ago
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,055Updated last year
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆826Updated last week
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆595Updated this week
• tensorflow / mlir-hlo
  ☆417Updated last week
• NVIDIA / cuCollections
  ☆552Updated this week
• NVIDIA / NVTX
  The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…
  ☆413Updated last week
• 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
• tensorflow / custom-op
  Guide for building custom op for TensorFlow
  ☆382Updated 2 years ago
• pytorch / tvm
  TVM integration into PyTorch
  ☆453Updated 5 years ago
• zdevito / ATen
  ATen: A TENsor library for C++11
  ☆699Updated 5 years ago
• tensor-compiler / taco
  The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
  ☆1,306Updated 2 months ago
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆679Updated this week
• d2l-ai / d2l-tvm
  Dive into Deep Learning Compiler
  ☆646Updated 3 years ago
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆497Updated last week
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆883Updated this week
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,577Updated last week
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆343Updated this week
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆503Updated 3 years ago
• pytorch / builder
  Continuous builder and binary build scripts for pytorch
  ☆353Updated 2 months ago
• NVIDIA / cub
  [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
  ☆1,757Updated last year
• nv-legate / cupynumeric
  An Aspiring Drop-In Replacement for NumPy at Scale
  ☆904Updated this week
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆707Updated 3 weeks ago
• inducer / loopy
  A code generator for array-based code on CPUs and GPUs
  ☆608Updated last week
• libxsmm / libxsmm
  Library for specialized dense and sparse matrix operations, and deep learning primitives.
  ☆881Updated last week
• ezyang / nvprof2json
  Convert nvprof profiles into about:tracing compatible JSON files
  ☆69Updated 4 years ago