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