Linux-cpp-lisp / opt_einsum_fxLinks
Einsum optimization using opt_einsum and PyTorch FX graph rewriting
☆22Updated 3 years ago
Alternatives and similar repositories for opt_einsum_fx
Users that are interested in opt_einsum_fx are comparing it to the libraries listed below
Sorting:
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- ☆52Updated last year
- Collection of scripts to build PyTorch and the domain libraries from source.☆12Updated last month
- cuASR: CUDA Algebra for Semirings☆39Updated 3 years ago
- Generic PyTorch implementation of einsum that supports different semirings☆50Updated last year
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆98Updated this week
- Compiler with automatic differentiation☆48Updated last year
- NPBench - A Benchmarking Suite for High-Performance NumPy☆89Updated last week
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- ☆106Updated last year
- OpTree: Optimized PyTree Utilities☆196Updated this week
- ☆14Updated 2 years ago
- ☆15Updated last year
- A tool for weighted model counting through tensor network contraction☆21Updated 3 years ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Updated 2 years ago
- Documentation:☆123Updated 2 years ago
- MLIR tools and dialect for GraphBLAS☆18Updated 3 years ago
- A Data-Centric Compiler for Machine Learning☆84Updated last year
- A library for syntactically rewriting Python programs, pronounced (sinner).☆68Updated 3 years ago
- JAX for Graphcore IPU (experimental)☆22Updated last year
- Julia ports of the Rodinia benchmark suite for heterogeneous computing infrastructures☆54Updated 2 years ago
- ☆28Updated 8 months ago
- Use numba-compiled kernels from within Jax☆31Updated 5 months ago
- Torch Frontend for IREE☆25Updated last year
- High-Performance Tensor Transpose library☆205Updated 2 years ago
- Cyclops Tensor Framework: parallel arithmetic on multidimensional arrays☆208Updated 3 months ago
- Flexible and performant GEMM kernels in Julia☆83Updated 2 months ago
- Automatic Differentiation for Tensor Algebras☆28Updated 7 years ago
- ☆61Updated 2 years ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 3 weeks ago