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:
- Collection of scripts to build PyTorch and the domain libraries from source.☆13Updated last month
- ☆54Updated last year
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- NPBench - A Benchmarking Suite for High-Performance NumPy☆91Updated 3 weeks ago
- cuASR: CUDA Algebra for Semirings☆42Updated 3 years ago
- Compiler with automatic differentiation☆49Updated 2 years ago
- ☆16Updated last year
- A tracing JIT compiler for PyTorch☆13Updated 4 years ago
- Generic PyTorch implementation of einsum that supports different semirings☆51Updated 3 weeks ago
- The Surprisingly ParalleL spArse Tensor Toolkit.☆73Updated 3 years ago
- Tensor Contraction Code Generator☆39Updated 8 years ago
- Strassen's Algorithm for Tensor Contraction☆14Updated 8 years ago
- Automatic Differentiation for Tensor Algebras☆28Updated 7 years ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆107Updated this week
- A library for syntactically rewriting Python programs, pronounced (sinner).☆68Updated 3 years ago
- CUDA and OpenMP implementations of C2R/R2C inplace transposition☆48Updated 10 years ago
- A Data-Centric Compiler for Machine Learning☆85Updated 2 weeks ago
- ☆14Updated 2 years ago
- ☆28Updated 11 months ago
- Cyclops Tensor Framework: parallel arithmetic on multidimensional arrays☆210Updated 6 months ago
- High-Performance Tensor Transpose library☆204Updated 2 years ago
- An MLIR-based source-to-source automatic differentiation system.☆14Updated 2 years ago
- Distributed Communication-Optimal Matrix-Matrix Multiplication Algorithm☆212Updated 3 weeks ago
- Documentation:☆125Updated 2 years ago
- A tool for weighted model counting through tensor network contraction☆21Updated 4 years ago
- Flexible and performant GEMM kernels in Julia☆83Updated last week
- The Combinatorial BLAS (CombBLAS) is an extensible distributed-memory parallel graph library offering a small but powerful set of linear …☆80Updated 4 months ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- Worked example of the process from Python source to CUDA kernel execution with Numba☆44Updated last year
- Generic API for dispatch to Pyro backends.☆16Updated 3 years ago