Stretching GPU performance for GEMMs and tensor contractions.
-
Updated
Jun 2, 2024 - Python
Stretching GPU performance for GEMMs and tensor contractions.
TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch
Provides compile-time contraction pattern analysis to determine optimal tensor operation to perform.
Julia package for tensor contractions and related operations
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
Hyper optimized contraction trees for large tensor networks and einsums
Machine Intelligence Shader Autogen. AMDGPU ML shader code generator. (previously iGEMMgen)
Rust accelerated contraction ordering primitives for tensor networks and einsums
Julia wrapper for TBLIS with TensorOperations.jl
A lightweight high performance tensor algebra framework for modern C++
einsum via batch matrix multiply
Transposition-free Complex Tensor Contractions
Database approach for generating near-roofline Einstein Summation kernels.
A Python utility for analyzing a given solution to the Einstein's field equations. Built on Sympy.
Collection of basic and advanced Tensor Algebra operations using Matlab and Python.
C++ Header-Only Library for High-Performance Tensor-Vector Multiplication
Tensor Contraction C++ Library
Strassen's Algorithm for Tensor Contraction
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."