⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
-
Updated
Oct 1, 2024 - Python
#
einsum
Here are 29 public repositories matching this topic...
One More Einsum for Julia! With runtime order-specification and high-level adjoints for AD
-
Updated
Nov 28, 2024 - Julia
Hyper optimized contraction trees for large tensor networks and einsums
-
Updated
Dec 20, 2024 - Python
Stats, linear algebra and einops for xarray
-
Updated
Nov 19, 2024 - Python
Provides compile-time contraction pattern analysis to determine optimal tensor operation to perform.
cpp matrix linear-algebra scientific-computing tensor tensors tensor-contraction tensor-decomposition matrix-library cp-decomposition matrix-computations cpp20 tucker-decomposition einsum dense-matrices
-
Updated
Nov 13, 2024 - C++
A generic implementation of tensor einsum in Fortran.
-
Updated
Jan 29, 2021 - Fortran
An implementation of EinsumNetworks in PyTorch.
-
Updated
Nov 11, 2024 - Jupyter Notebook
Einsum optimization using opt_einsum and PyTorch FX graph rewriting
-
Updated
Mar 17, 2022 - Python
Memory-efficient optimum einsum using opt_einsum planning and PyTorch kernels.
-
Updated
Apr 24, 2023 - Python
Rust accelerated contraction ordering primitives for tensor networks and einsums
-
Updated
Dec 23, 2024 - Rust
Convolutions and more as einsum for PyTorch
fisher pytorch convolution ntk transpose-convolution im2col kfac einsum tensornetwork einsum-notation vjp jvp ggn
-
Updated
Jun 6, 2024 - Python
einsum via batch matrix multiply
-
Updated
Nov 29, 2023 - Python
Einsum Expressions in Julia
-
Updated
Nov 24, 2024 - TeX
A Tensor Network package for Machine Learning and Quantum Computing in Python.
algebra quantum-computing tensor matrix-product-states tensor-train renormalization-group tensor-network einsum matrix-product-operators
-
Updated
Mar 20, 2022 - Python
Optimizing Einstein Sum Implementation in PyTorch with Specialization, Path Searching, and GPU-Native Contraction
benchmarking performance pytorch dynamic-programming profiling performance-optimization einsum cutensor einsum-notation
-
Updated
Sep 25, 2020 - Python
A collection of state-of-the-art contraction ordering algorithms. https://arxiv.org/abs/2209.12332
dynamic-programming einsum tensornetwork opt-einsum join-ordering contraction-order optimal-algorithm cotengra
-
Updated
Dec 3, 2024 - Python
Improve this page
Add a description, image, and links to the einsum topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the einsum topic, visit your repo's landing page and select "manage topics."