⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
-
Updated
May 27, 2024 - Python
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
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
Machine Intelligence Shader Autogen. AMDGPU ML shader code generator. (previously iGEMMgen)
einsum via batch matrix multiply
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."