tensor-contraction

There are 1 repository under tensor-contraction topic.

dgasmith / opt_einsum
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
python tensor contraction gpu-acceleration performance einsum tensor-contraction
Language:Python 803
romeric / Fastor
A lightweight high performance tensor algebra framework for modern C++
tensors tensor-contraction multidimensional-arrays simd hpc small-blas fpga
Language:C++ 699
TensorOperations.jl
Jutho / TensorOperations.jl
Julia package for tensor contractions and related operations
tensor tensor-contraction tensor-permutation tensor-trace tensor-transposition tensor-operations einstein-summation index-notation
Language:Julia 412
ROCm / Tensile
Stretching GPU performance for GEMMs and tensor contractions.
gemm blas dnn neural-networks machine-learning tensors python opencl hip radeon auto-tuning radeon-open-compute amd gpu-computing gpu-acceleration gpu matrix-multiplication tensor-contraction assembly
Language:Python 189
cotengra
jcmgray / cotengra
Hyper optimized contraction trees for large tensor networks and einsums
tensor-networks tensor-contraction tensor-network quimb opt-einsum einsum tensor
Language:Python 165
spacetimeengineer / spacetimeengine
A Python utility for analyzing a given solution to the Einstein's field equations. Built on Sympy.
gravity gravity-simulation spacetime einstein-field-equations einstein-coefficients sympy ricci-curvature riemannian-geometry riemann connection metric-theory tensor-contraction tensor-algebra einstein computer-algebra science-research research-tool spacetime-toolkit
Language:Python 143
tensorly / torch
TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch
deep-learning deep-neural-networks deep-tensor-learning factorized-cnns factorized-convolution factorized-networks pytorch tensor tensor-contraction tensor-convolution-networks tensor-learning tensor-networks tensor-regression tensor-regression-layers tensorized-networks
Language:Python 67
springer13 / tcl
Tensor Contraction C++ Library
high-performance tensor tensor-contraction cpp11 multidimensional-arrays python
Language:C++ 50
Einsums / Einsums
Provides compile-time contraction pattern analysis to determine optimal tensor operation to perform.
einsum cpp tensor dense-matrices linear-algebra matrix matrix-computations matrix-library cp-decomposition tensor-decomposition tucker-decomposition tensor-contraction scientific-computing tensors cpp20
Language:C++ 31
ROCm / MISA
Machine Intelligence Shader Autogen. AMDGPU ML shader code generator. (previously iGEMMgen)
amd assembly convolution gpu implicit-gemm implicit-gemm-algorithm python tensor-contraction
Language:Python 31
bassoy / ttv
C++ Header-Only Library for High-Performance Tensor-Vector Multiplication
arrays blas c-plus-plus fast high-performance multidimensional multilinear-algebra tensor tensor-contraction tensor-library tensor-times-vector tensor-vector-multiplication tensor-vector-multiplications
Language:C++ 19
KennethBenicio / MSc-Multilinear-Algebra
Collection of basic and advanced Tensor Algebra operations using Matlab and Python.
folding khatri-rao kronecker-product least-squares matlab multilinear-algebra multilinear-svd parafac python tensor-algebra tensor-operations unfolding als alternating-least-squares cpd tensor-product mathematics tensor-contraction tensor-train
Language:Jupyter Notebook 13
jcmgray / cotengrust
Rust accelerated contraction ordering primitives for tensor networks and einsums
contraction-order einsum tensor-networks contraction tensor tensor-contraction tensor-network
Language:Rust 10
jcmgray / einsum_bmm
einsum via batch matrix multiply
einsum tensor tensor-contraction tensor-networks
Language:Python 10
flame / tblis-strassen
Strassen's Algorithm for Tensor Contraction
tensor-contraction strassen-algorithm strassen
Language:C 9
MatthiasReumann / t1m
Transposition-free Complex Tensor Contractions
tensors complex-tensors tensor tensor-contraction
Language:C++ 5
kaushikcfd / feinsum
Database approach for generating near-roofline Einstein Summation kernels.
code-generation einstein-summation linear-algebra tensor-contraction multidimensional-arrays python sql tensors gpu opencl
Language:Python 3