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:
- ☆52Updated last year
- Collection of scripts to build PyTorch and the domain libraries from source.☆12Updated last week
- Compiler with automatic differentiation☆48Updated 2 years ago
- Generic PyTorch implementation of einsum that supports different semirings☆50Updated last year
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- NPBench - A Benchmarking Suite for High-Performance NumPy☆89Updated 3 weeks ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆100Updated this week
- ☆14Updated 2 years ago
- ☆16Updated last year
- cuASR: CUDA Algebra for Semirings☆39Updated 3 years ago
- Flexible and performant GEMM kernels in Julia☆83Updated 3 months ago
- A tracing JIT compiler for PyTorch☆13Updated 3 years ago
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.☆148Updated 2 years ago
- MLIR tools and dialect for GraphBLAS☆18Updated 3 years ago
- Sparsity support for PyTorch☆37Updated 7 months ago
- A tool for weighted model counting through tensor network contraction☆21Updated 3 years ago
- An Aspiring Drop-In Replacement for Pandas at Scale☆74Updated 4 years ago
- CUDA templates for tile-sparse matrix multiplication based on CUTLASS.☆50Updated 7 years ago
- ☆63Updated 8 months ago
- Python bindings for the egglog rust library☆90Updated last week
- Automatic Differentiation for Tensor Algebras☆28Updated 7 years ago
- Implementation of a Tensorflow XLA rematerialization pass☆15Updated 5 years ago
- An MLIR-based source-to-source automatic differentiation system.☆14Updated 2 years ago
- A Data-Centric Compiler for Machine Learning☆85Updated last year
- Julia ports of the Rodinia benchmark suite for heterogeneous computing infrastructures☆54Updated 2 years ago
- Tokamax: A GPU and TPU kernel library.☆95Updated this week
- SMT-LIB benchmarks for shape computations from deep learning models in PyTorch☆18Updated 2 years ago
- SParse AcceleRation on Tensor Architecture☆17Updated 6 months ago
- Tensor Contraction Code Generator☆39Updated 8 years ago
- ☆107Updated last year