Stretching GPU performance for GEMMs and tensor contractions.
-
Updated
Aug 10, 2024 - Python
Stretching GPU performance for GEMMs and tensor contractions.
Hyper optimized contraction trees for large tensor networks and einsums
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
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.
A Python utility for analyzing a given solution to the Einstein's field equations. Built on Sympy.
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."