Search Packages

cotengra

Hyper optimized contraction trees for large tensor networks and einsums

tensorly-torch

TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch

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.

spacetimeengine

A Python utility for analyzing a given solution to the Einstein's field equations. Built on Sympy.

feinsum

Fast(er?) Einsum