Alternatives and similar repositories for dlpack:

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

• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆695Updated 2 years ago
• tensorflow / mlir-hlo
  ☆403Updated this week
• facebookincubator / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,255Updated 3 weeks ago
• pytorch / tvm
  TVM integration into PyTorch
  ☆453Updated 5 years ago
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,246Updated this week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,019Updated 9 months ago
• tensorflow / runtime
  A performant and modular runtime for TensorFlow
  ☆759Updated this week
• NVIDIA / cuCollections
  ☆506Updated this week
• microsoft / nnfusion
  A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
  ☆973Updated 4 months ago
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆252Updated 2 years ago
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆436Updated this week
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆759Updated last week
• d2l-ai / d2l-tvm
  Dive into Deep Learning Compiler
  ☆645Updated 2 years ago
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆525Updated this week
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆550Updated 2 months ago
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆801Updated 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…
  ☆332Updated last week
• inducer / loopy
  A code generator for array-based code on CPUs and GPUs
  ☆596Updated this week
• zdevito / ATen
  ATen: A TENsor library for C++11
  ☆690Updated 5 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆500Updated 3 years ago
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆324Updated last month
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,410Updated this week
• NVIDIA / gdrcopy
  A fast GPU memory copy library based on NVIDIA GPUDirect RDMA technology
  ☆932Updated last month
• NVIDIA / jitify
  A single-header C++ library for simplifying the use of CUDA Runtime Compilation (NVRTC).
  ☆520Updated 8 months ago
• NVIDIA / cub
  [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
  ☆1,694Updated last year
• baidu-research / baidu-allreduce
  ☆578Updated 6 years ago
• tensor-compiler / taco
  The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
  ☆1,271Updated 9 months ago
• gunrock / gunrock
  Programmable CUDA/C++ GPU Graph Analytics
  ☆1,002Updated 5 months ago
• libxsmm / libxsmm
  Library for specialized dense and sparse matrix operations, and deep learning primitives.
  ☆859Updated this week
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆672Updated last week