Optimizing Einstein Sum Implementation in PyTorch with Specialization, Path Searching, and GPU-Native Contraction
-
Updated
Sep 25, 2020 - Python
Optimizing Einstein Sum Implementation in PyTorch with Specialization, Path Searching, and GPU-Native Contraction
A generic implementation of tensor einsum in Fortran.
Personal fork of the official EinsumNetworks implementation with a few enhancements.
Einsum optimization using opt_einsum and PyTorch FX graph rewriting
A Tensor Network package for Machine Learning and Quantum Computing in Python.
Julia rewrite of opt_einsum Python package
Memory-efficient optimum einsum using opt_einsum planning and PyTorch kernels.
A collection of state-of-the-art contraction ordering algorithms. https://arxiv.org/abs/2209.12332
Convolutions and more as einsum for PyTorch
einsum via batch matrix multiply
MyPy Type Checking for NumPy/Jax/PyTorch Einsum Operations
Stats, linear algebra and einops for xarray
Add a description, image, and links to the einsum topic page so that developers can more easily learn about it.
To associate your repository with the einsum topic, visit your repo's landing page and select "manage topics."