dmlc / dlpack

Related projects ⓘ

Alternatives and complementary repositories for dlpack

• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,202Updated this week
• pytorch / tvm
  TVM integration into PyTorch
  ☆453Updated 4 years ago
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆691Updated last year
• tensorflow / runtime
  A performant and modular runtime for TensorFlow
  ☆756Updated 3 weeks ago
• facebookincubator / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,218Updated this week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,010Updated 6 months ago
• microsoft / nnfusion
  A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
  ☆959Updated last month
• d2l-ai / d2l-tvm
  Dive into Deep Learning Compiler
  ☆641Updated 2 years ago
• NVIDIA / cuCollections
  ☆482Updated this week
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆247Updated last year
• tensorflow / mlir-hlo
  ☆398Updated this week
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆727Updated this week
• inducer / loopy
  A code generator for array-based code on CPUs and GPUs
  ☆587Updated this week
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆492Updated this week
• nv-legate / cunumeric
  An Aspiring Drop-In Replacement for NumPy at Scale
  ☆620Updated last month
• tensorflow / custom-op
  Guide for building custom op for TensorFlow
  ☆378Updated last year
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆512Updated 2 weeks ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆493Updated 3 years ago
• tensor-compiler / taco
  The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
  ☆1,254Updated 6 months ago
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆406Updated this week
• zdevito / ATen
  ATen: A TENsor library for C++11
  ☆683Updated 4 years ago
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,348Updated 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…
  ☆301Updated this week
• libxsmm / libxsmm
  Library for specialized dense and sparse matrix operations, and deep learning primitives.
  ☆850Updated this week
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆760Updated this week
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆649Updated this week
• pytorch / multipy
  torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…
  ☆174Updated this week
• NVIDIA / cub
  [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
  ☆1,679Updated last year
• NVIDIA / jitify
  A single-header C++ library for simplifying the use of CUDA Runtime Compilation (NVRTC).
  ☆517Updated 5 months ago
• rapidsai / raft
  RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-a…
  ☆766Updated this week