Stretching GPU performance for GEMMs and tensor contractions.
-
Updated
Jul 13, 2024 - Python
Stretching GPU performance for GEMMs and tensor contractions.
A lightweight high performance tensor algebra framework for modern C++
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
Julia package for tensor contractions and related operations
Hyper optimized contraction trees for large tensor networks and einsums
TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch
A Python utility for analyzing a given solution to the Einstein's field equations. Built on Sympy.
Tensor Contraction C++ Library
Machine Intelligence Shader Autogen. AMDGPU ML shader code generator. (previously iGEMMgen)
Provides compile-time contraction pattern analysis to determine optimal tensor operation to perform.
Strassen's Algorithm for Tensor Contraction
C++ Header-Only Library for High-Performance Tensor-Vector Multiplication
Collection of basic and advanced Tensor Algebra operations using Matlab and Python.
Database approach for generating near-roofline Einstein Summation kernels.
Transposition-free Complex Tensor Contractions
Julia wrapper for TBLIS with TensorOperations.jl
Rust accelerated contraction ordering primitives for tensor networks and einsums
einsum via batch matrix multiply
Add a description, image, and links to the tensor-contraction topic page so that developers can more easily learn about it.
To associate your repository with the tensor-contraction topic, visit your repo's landing page and select "manage topics."