PyRank
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About

Einsum Python Packages

Python packages with the GitHub topic einsum. Sorted by relevance, with stars and monthly downloads.
arviz-devs
xarray-einstats

Stats, linear algebra and einops for xarray

2.3M 68 9
Linux-cpp-lisp
opt-einsum-fx

Einsum optimization using opt_einsum and PyTorch FX graph rewriting

1.2M 22 5
jcmgray
cotengra

Hyper optimized contraction trees for large tensor networks and einsums

161K 241 36
f-dangel
einconv

Convolutions and more as einsum for PyTorch

42K 18 1
bdusell
torch-semiring-einsum

Generic PyTorch implementation of einsum that supports different semirings

6K 50 8
DOKOS-TAYOS
tensor-network-editor

Interactive local editor for building tensor networks visually and generating readable Python code for tensornetwork, quimb, tensorkrowch, and einsum.

2K 2 0
DOKOS-TAYOS
tensor-network-visualization

2D and 3D tensor network visualization, designed for torch, numpy, quimb, tensorkrowch, tenpy and tensornetwork compatibility

2K 1 0
trnsci
trntensor

Tensor contractions for AWS Trainium via NKI (cuTENSOR-equivalent) — einsum with contraction planning, CP/PARAFAC and Tucker decompositions, density-fitted post-Hartree-Fock patterns.

282 0 0
hhaoyan
opt-einsum-torch

Memory-efficient optimum einsum using opt_einsum planning and PyTorch kernels.

206 16 2
adtzlr
neinsum

NumPy's Einsum, but with named subscripts.

134 2 0
hzhangxyz
easy-ast

This package `easy-ast` contains several utility about AST transformer.

105 0 0
Amedar-Asterisk
u-stats

A Python package for U-Statistics calculation

101 4 0
lgeiger
mypy-einsum

MyPy Type Checking for NumPy/Jax/PyTorch Einsum Operations

72 1 0
guilt
python-einsum

Simple, readable Einstein Summation in Python.

34 0 0
    • Data from PyPI, GitHub, ClickHouse, and BigQuery