Linux-cpp-lisp / opt_einsum_fxLinks
Einsum optimization using opt_einsum and PyTorch FX graph rewriting
☆21Updated 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:
- Collection of scripts to build PyTorch and the domain libraries from source.☆11Updated last week
- ☆52Updated 10 months ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- cuASR: CUDA Algebra for Semirings☆36Updated 2 years ago
- Compiler with automatic differentiation☆48Updated last year
- Flexible and performant GEMM kernels in Julia☆82Updated this week
- Generic PyTorch implementation of einsum that supports different semirings☆49Updated 11 months ago
- Worked example of the process from Python source to CUDA kernel execution with Numba☆41Updated 9 months ago
- CUDA and OpenMP implementations of C2R/R2C inplace transposition☆46Updated 10 years ago
- ☆62Updated 4 months ago
- A tool for weighted model counting through tensor network contraction☆21Updated 3 years ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- ☆15Updated 2 years ago
- MLIR tools and dialect for GraphBLAS☆18Updated 3 years ago
- Automatic High-Order Optimization for Tensors☆23Updated 2 years ago
- ☆28Updated 3 years ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆86Updated this week
- ☆16Updated 9 months ago
- Generic API for dispatch to Pyro backends.☆16Updated 3 years ago
- A tracing JIT for PyTorch☆17Updated 2 years ago
- Code accompanying the paper "Automating Involutive MCMC using Probabilistic and Differentiable Programming"☆13Updated 4 years ago
- Implementation of a Tensorflow XLA rematerialization pass☆15Updated 5 years ago
- ☆19Updated last month
- Recursive LAPACK Collection☆42Updated 3 years ago
- ☆57Updated last year
- Collection of simple General Matrix Multiplication - GEMM implementations☆13Updated last year
- A tracing JIT compiler for PyTorch☆13Updated 3 years ago
- JAX for Graphcore IPU (experimental)☆21Updated last year
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Updated 2 years ago
- ☆28Updated 5 months ago