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:
- ☆52Updated last year
- Collection of scripts to build PyTorch and the domain libraries from source.☆12Updated 3 weeks ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Generic PyTorch implementation of einsum that supports different semirings☆50Updated last year
- NPBench - A Benchmarking Suite for High-Performance NumPy☆87Updated 4 months ago
- cuASR: CUDA Algebra for Semirings☆38Updated 3 years ago
- Compiler with automatic differentiation☆48Updated last year
- A tool for weighted model counting through tensor network contraction☆21Updated 3 years ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆95Updated this week
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.☆148Updated 2 years ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- ☆16Updated 11 months 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
- An Aspiring Drop-In Replacement for Pandas at Scale☆74Updated 3 years ago
- Sparsity support for PyTorch☆37Updated 5 months ago
- Linnea is an experimental tool for the automatic generation of optimized code for linear algebra problems.☆71Updated 3 weeks ago
- A tracing JIT compiler for PyTorch☆13Updated 3 years ago
- High dimensional black-box optimizer using Latent Action Monte Carlo Tree Search algorithm☆29Updated 3 years ago
- ☆105Updated last year
- OpTree: Optimized PyTree Utilities☆192Updated this week
- ☆28Updated 3 years ago
- Flexible and performant GEMM kernels in Julia☆83Updated last month
- Cyclops Tensor Framework: parallel arithmetic on multidimensional arrays☆208Updated 3 months ago
- Worked example of the process from Python source to CUDA kernel execution with Numba☆42Updated last year
- A Data-Centric Compiler for Machine Learning☆84Updated last year
- POC work on MLIR backend☆59Updated last year
- This repository is the summary of all of our works for the XLA.☆11Updated 7 years ago
- FP64 equivalent GEMM via Int8 Tensor Cores using the Ozaki scheme☆86Updated 5 months ago