tensor-compiler / taco
The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
☆1,271Updated 9 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.☆859Updated this week
- Programmable CUDA/C++ GPU Graph Analytics☆1,002Updated 5 months ago
- common in-memory tensor structure☆930Updated 4 months ago
- A domain specific language to express machine learning workloads.☆1,758Updated last year
- High-performance automatic differentiation of LLVM and MLIR.☆1,325Updated this week
- [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl☆1,694Updated last year
- The Legion Parallel Programming System☆703Updated 3 weeks ago
- Patterns and behaviors for GPU computing☆1,691Updated 2 years ago
- Collective communications library with various primitives for multi-machine training.☆1,255Updated 3 weeks ago
- The Tensor Algebra SuperOptimizer for Deep Learning☆695Updated 2 years ago
- nGraph has moved to OpenVINO☆1,349Updated 4 years ago
- "Multi-Level Intermediate Representation" Compiler Infrastructure☆1,735Updated 3 years ago
- ATen: A TENsor library for C++11☆690Updated 5 years ago
- A single-header C++ library for simplifying the use of CUDA Runtime Compilation (NVRTC).☆520Updated 8 months ago
- A code generator for array-based code on CPUs and GPUs☆596Updated this week
- Backward compatible ML compute opset inspired by HLO/MHLO☆436Updated this week
- Assembler for NVIDIA Maxwell architecture☆964Updated 2 years ago
- CUSP : A C++ Templated Sparse Matrix Library☆409Updated 2 months ago
- ☆506Updated this week
- ☆403Updated this week
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,410Updated this week
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,243Updated this week
- Low-precision matrix multiplication☆1,790Updated last year
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,246Updated this week
- Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure☆801Updated this week
- oneAPI Deep Neural Network Library (oneDNN)☆3,694Updated this week
- Distributed multigrid linear solver library on GPU☆519Updated 5 months ago
- A lightweight high performance tensor algebra framework for modern C++☆764Updated 9 months ago
- CUDA Core Compute Libraries☆1,401Updated this week
- ☆1,656Updated 6 years ago