tensor-compiler / tacoLinks
The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
☆1,347Updated 9 months ago
Alternatives and similar repositories for taco
Users that are interested in taco are comparing it to the libraries listed below
Sorting:
- Library for specialized dense and sparse matrix operations, and deep learning primitives.☆933Updated 3 weeks ago
- common in-memory tensor structure☆1,154Updated last week
- High-performance automatic differentiation of LLVM and MLIR.☆1,531Updated this week
- A code generator for array-based code on CPUs and GPUs☆622Updated last week
- Programmable CUDA/C++ GPU Graph Analytics☆1,063Updated this week
- ATen: A TENsor library for C++11☆713Updated 6 years ago
- The Legion Parallel Programming System☆751Updated last month
- [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl☆1,817Updated 2 years ago
- Patterns and behaviors for GPU computing☆1,762Updated 2 weeks ago
- The Tensor Algebra SuperOptimizer for Deep Learning☆736Updated 3 years ago
- A domain specific language to express machine learning workloads.☆1,765Updated 2 years ago
- nGraph has moved to OpenVINO☆1,346Updated 5 years ago
- "Multi-Level Intermediate Representation" Compiler Infrastructure☆1,763Updated 4 years ago
- Low-precision matrix multiplication☆1,829Updated 2 years ago
- Tuned OpenCL BLAS☆1,162Updated 2 months ago
- Collective communications library with various primitives for multi-machine training.☆1,390Updated 2 weeks ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,520Updated this week
- A single-header C++ library for simplifying the use of CUDA Runtime Compilation (NVRTC).☆569Updated 4 months ago
- Source code examples from the Parallel Forall Blog☆1,321Updated 4 months ago
- CUSP : A C++ Templated Sparse Matrix Library☆421Updated 5 months ago
- An efficient C++20 GPU numerical computing library with Python-like syntax☆1,400Updated this week
- ☆615Updated this week
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,740Updated this week
- Assembler for NVIDIA Maxwell architecture☆1,060Updated 3 years ago
- A performant and modular runtime for TensorFlow☆754Updated 4 months ago
- Kernel Tuner☆381Updated this week
- A lightweight high performance tensor algebra framework for modern C++☆828Updated 6 months ago
- CUDA Data Parallel Primitives Library☆438Updated 7 years ago
- Backward compatible ML compute opset inspired by HLO/MHLO☆598Updated 2 weeks ago
- Acceleration package for neural networks on multi-core CPUs☆1,703Updated last year