tensor-compiler / taco
The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
☆1,285Updated 10 months ago
Alternatives and similar repositories for taco:
Users that are interested in taco are comparing it to the libraries listed below
- Library for specialized dense and sparse matrix operations, and deep learning primitives.☆865Updated this week
- Programmable CUDA/C++ GPU Graph Analytics☆1,012Updated 7 months ago
- [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl☆1,731Updated last year
- Patterns and behaviors for GPU computing☆1,705Updated 2 years ago
- common in-memory tensor structure☆957Updated 2 weeks ago
- A code generator for array-based code on CPUs and GPUs☆597Updated this week
- The Legion Parallel Programming System☆708Updated last week
- ATen: A TENsor library for C++11☆693Updated 5 years ago
- A single-header C++ library for simplifying the use of CUDA Runtime Compilation (NVRTC).☆527Updated 3 weeks ago
- The Tensor Algebra SuperOptimizer for Deep Learning☆703Updated 2 years ago
- A domain specific language to express machine learning workloads.☆1,756Updated last year
- Thin, unified, C++-flavored wrappers for the CUDA APIs☆825Updated 2 weeks ago
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,276Updated this week
- ☆520Updated this week
- Collective communications library with various primitives for multi-machine training.☆1,273Updated this week
- CUSP : A C++ Templated Sparse Matrix Library☆411Updated 4 months ago
- A lightweight high performance tensor algebra framework for modern C++☆773Updated 10 months ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,270Updated this week
- High-performance automatic differentiation of LLVM and MLIR.☆1,349Updated this week
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,462Updated this week
- A performant and modular runtime for TensorFlow☆759Updated 2 weeks ago
- CUDA Kernel Benchmarking Library☆585Updated 3 months ago
- Source code examples from the Parallel Forall Blog☆1,267Updated 7 months ago
- Assembler for NVIDIA Maxwell architecture☆977Updated 2 years ago
- "Multi-Level Intermediate Representation" Compiler Infrastructure☆1,738Updated 3 years ago
- BLISlab: A Sandbox for Optimizing GEMM☆503Updated 3 years ago
- Numerical linear algebra software package☆440Updated this week
- CUDA Core Compute Libraries☆1,509Updated this week
- Low-precision matrix multiplication☆1,792Updated last year
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆979Updated 5 months ago